YOLOv5

2022-04-07

install:: git clone https://github.com/ultralytics/yolov5.git

github:: ultralytics/yolov5

docs:: YOLOv5 Docs

预训练模型

下载地址：ultralytics/yolov5

  • P5 模型
    • yolov5s.pt：速度最快，网络参数最少，效果最差
    • yolov5m.pt
    • yolov5l.pt
    • yolov5x.pt：速度慢，参数多，效果好
  • P6 模型（不带6的为 P5）
    • yolov5s6.pt
    • yolov5m6.pt
    • yolov5l6.pt
    • yolov5x6.pt

文件目录说明

  • data: 数据集配置，自己训练的数据集会用到
  • model: 参数模型，训练时会用到五个 yaml 文件
  • run：存放检测模型的效果图
  • detect.py: 测试模型效果
  • train.py：训练模型

detect.py

  • —weights: 指定测试模型，默认为 yolov5s.pt
  • —source: 测试数据来源，支持图片、视频、 [[YouTube]]、[[RTSP]] 等
  • —imgsz: 测试图片大小，一般默认
  • —conf-thres: 置信度，大于指定数值就认为检测成功
  • —iou-thres: 交并比
  • —view-img: 边检测边输出效果
  • save-txt: 是否保留预测坐标，默认 False
  • nosave: 不保存图片或视频，默认 False
  • exist-ok: 不自动增加序号
  • save-crop: 按 name 独立保存每个结果为图片，默认 False

变量说明

  • label: 保存模型类别名称和对应置信度（person 0.85）
  • xyxy: 第二个是 box 左上角的高度
  • conf: 置信度
  • cls: names 的对应编号

图像标注软件

测试

支持网络摄像头、图片、视频、目录（指定类型）和链接（YouTube、RTSP、RTMP、HTTP 等）

Terminal window
1
# 本机摄像头
2
python detecty.py --weights yolov5s.py --source 0
3


4
# RTSP
5
python detecty.py --weights yolov5s.py --source rtsp://10.1.1.19:554/Streamaing/Channels/101

参考

# AI
# 视觉
# YOLO
# Ultralytics