用yolov5训练自己的模型并测试

训练模型

1、首先将数据集放在根目录下,名字一定要是imageslabels,里面的目录名字最好为trainval,要不然系统会判别不出来
用yolov5训练自己的模型并测试_第1张图片
2、然后复制一份coco128改名字为mydata.yaml,然后修改里面的内容
用yolov5训练自己的模型并测试_第2张图片
3、最后修改train.py文件第三行的参数为mydata.yaml,让他指定自己的训练数据

def parse_opt(known

你可能感兴趣的:(python,pytorch)