Window10使用Pytorch+Yolov5進行訓練

Step1)執行以下命令進行訓練，並命名為coco128

#python train.py --img 640 --batch 8 --epochs 3 --data coco128.yaml --weights yolov5s.pt --name coco128

使用yolov5專案內的coco128資料集(coco128.yaml，如下)，訓練時會自動下載coco128資料集與權重檔yolov5s.pt，到上一層目錄的datasets資料夾。

執行時會下載資料集coco128，如下圖。

訓練3個Epoch後，訓練後的權重檔儲存在runs\train\coco128\weights。