YoloV5模型训练

数据调整

../   <=项目根目录
├── images   <=图片目录
│   ├── train   <=训练图片目录
│   └── val   <=验证图片目录
├── labels   <=标签目录
│   ├── train   <=训练集标签文件，要与训练集图片名称一一对应
│   └── val   <=证集标签文件，要与验证集图片名称一一对应
└── 其他

关键参数

weights:预训练的权重文件 data:数据集描述文件

数据训练

AttributeError: 'IterableList' object has no attribute 'origin'

结果分析

使用模型进行训练

python detect.py --weights runs/train/exp6/weights/best.pt --source mydemo/catandmouse.mp4 --view-img