afangyana的博客

原创 跑通基于YOLOv5的旋转框目标检测

2、编辑train.py ：workers最好设置为0 ，weights设为空（如果嫌浪费时间，可以加入预训练模型），cfg是关于模型参数的一些设置（在models/hub下可以看到一些已经给的参数设置，复制yolov5s.yaml路径到cfg中），data为指定数据集（也就是刚刚创建的my_data.yaml），对应的default填其路径，在这里应为data/my_data.yaml。在安装依赖包之前，需要安装与pytorch相对应的cuda版本，此项目使用的pytorch为1.7.0。