yolov5训练与验证的指令

yolov5训练与验证的指令

确认使用anaconda\envs下面的python.exe运行脚本

 

D:\anaconda_install\envs\pytorch\python.exe D:\zzy\yolov5-master\train.py --weights yolov5s1.pt --data D:\zzy\yolov5-master\data\side_defect.yaml --workers 1 --epoch 3 --batch-size 8 --device 0

注:--device 0 是gpu

--device cpu,是cpu

验证：

D:\anaconda_install\envs\pytorch\python.exe D:/zzy/yolov5-master/detect.py --weights best.pt --source D:\zzy\side_defect2\images\val