插件市场/YOLO 目标检测插件

YOLO 目标检测插件

基于 YOLO 深度学习模型,为 Hertz System 提供离线/批量图片和视频中的目标检测能力,可用于安全监控、商品识别、工业检测等场景。

分类:AI 检测 · 官方示例示例评分:4.9 ★ · 预计 1.2k+ 安装

适用版本:Hertz System v1.x

插件简介

本插件基于后端模块 hertz_studio_django_yolo,为 Hertz System 提供统一的目标检测服务接口,支持对图片或离线视频进行批量检测,并将检测结果与业务数据联动展示。

  • 支持对单张或批量图片进行目标检测,返回类别、置信度与边界框坐标;
  • 支持对离线视频进行逐帧检测,并导出检测结果或生成标注视频;
  • 可配置检测类别、置信度阈值、NMS 等参数,以兼顾精度与性能;
  • 可与业务模块联动,将检测到的对象绑定到工单、告警或统计报表中。

安装步骤

1. 后端安装(Django)

在部署 Hertz System 的后端虚拟环境中,通过公司私有 PyPI 源安装 YOLO 检测插件依赖:

pip install  hertz-studio-django-yolo -i https://username:password@hzpypi.hzsystems.cn/simple/

安装完成后,在项目的 settings.py 中注册应用:

INSTALLED_APPS += [
    "hertz_studio_django_yolo",
]

根据实际实现情况,执行数据库迁移或初始化脚本(如有需要):

python manage.py migrate

2. 前端集成

在管理端菜单中增加「YOLO 目标检测」入口,一般可放在「AI 能力」或「智能检测」分组下,指向目标检测任务管理页面。

前端通过统一的检测接口(例如 /api/ai/yolo/detect/),上传图片或选择已存在的文件,对检测结果进行可视化展示(边框、高亮、标签等)。

3. 配置参数示例

可以在配置中心或环境变量中设置 YOLO 检测相关参数,例如:

YOLO_MODEL_PATH=/opt/models/yolo.pt
YOLO_CONF_THRESHOLD=0.5
YOLO_IOU_THRESHOLD=0.45

上述配置仅为示例,具体值请根据实际模型文件位置和业务精度/性能要求进行调整。

使用教程

1. 创建检测任务

在「YOLO 目标检测」页面中创建检测任务,选择待检测的图片/视频资源、检测类别和阈值等参数,然后提交任务进行处理。

2. 查看检测结果

处理完成后,在任务详情页中查看检测结果,包括检测框、类别标签、置信度等,可支持对结果进行筛选与导出。

3. 与业务联动(可选)

可将检测结果与告警中心、工单管理等模块打通,例如对检测到的异常对象自动创建告警或工单,提升自动化程度。

效果预览

以下为 YOLO 目标检测插件在系统中的部分界面示意,包括图片目标检测结果展示和批量任务列表等页面。

YOLO 目标检测 - 图片检测结果示例 YOLO 目标检测 - 模型管理展示示例

将上述示例替换为你实际系统中的检测结果截图,点击图片可进入放大预览模式。

演示视频