YOLOv5

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 的对应编号

图像标注软件 §

• Make Sense#在线

测试 §

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

# 本机摄像头
python detecty.py --weights yolov5s.py --source 0
 
# RTSP
python detecty.py --weights yolov5s.py --source rtsp://10.1.1.19:554/Streamaing/Channels/101

参考 §

• Load YOLOv5 from PyTorch Hub ⭐ · Issue #36 · ultralytics/yolov5 · GitHub torch