YOLO v5跑通

YOLO v5 实现目标检测（参考数据集&自制数据集） - 知乎 (zhihu.com)

运行detect.py:

数据集：

coco128数据集是COCO数据集的前128张图片，通常用作小型的教程数据集，这里面的128张图片即用作训练也用作验证。

运行train.py：

运行时 --img 320 --batch-size 16

默认数值 batch_size=16, imgsz=640会报错outofmemory。

 python train.py --img 320 --batch-size 16 --epochs 5 --data ./data/coco128.yaml --cfg ./models/yolov5s.yaml --weights ./yolov5s.pt

根据cmd提示在runs/train/exp中查看验证集图片标记结果。