YOLO3-入門2-訓練-coco128
YOLO3 訓練
YOLO3的GitHub地址:
//github.com/ultralytics/yolov3github.com
部署平台:
//cloud.videojj.com
再回顧一下之前的安裝,連接實例後,在root目錄下:
git clone //github.com/ultralytics/yolov3 cd yolov3 pip install -r requirements.txt
選擇的數據集——COCO128
COCO128是一個小型教程數據集,由COCOtrain2017中的前128個影像組成。
數據集使用相同的128個影像進行訓練和測試。
它旨在通過在訓練較大的數據集之前演示對小數據集的過度擬合來驗證訓練管道。
在YOLOV3中自帶的coco128.yaml文件:
1)可選的用於自動下載的下載命令/URL,
2)指向培訓影像目錄的路徑(或指向帶有培訓影像列表的*.txt文件的路徑),
3)與驗證影像相同,
4)類數,
5)類名列表:
# download command/URL (optional) download: //github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip # train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/] train: ../coco128/images/train2017/ val: ../coco128/images/train2017/ # number of classes nc: 80 # class names names: ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light', 'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush']
訓練
Python train.py --img 640 --batch 4 --epochs 5 --data coco128.yaml --weights yolov3.pt其中–weights yolov3.pt文件,可以提前用wget指令下載好,避免運行train.py文件時,因網路原因造成錯誤。
wget //github.com/ultralytics/yolov3/releases/download/v9.5.0/yolov3.pt
其中–data coco128.yaml的coco128.yaml,是預裝在YOLOV3的程式碼中的。
路徑為:/root/yolov3/data。這裡還有很多其他的配置文件,可按需取用。

其中,batch的大小,epoch的大小,可以根據租用的機器的性能參數等,進行調整。
想要了解更多關於batch、epoch的,可以自己去搜一些部落格去學習。
部落客本人也不是特別懂,還在學習中。
訓練過程



訓練結果
運行結束後,會發現root路徑下,還有一個coco128的文件夾。 這是在coco128.yaml文件中有可選的用於自動下載的下載命令/URL。

在yolov3/runs/train/exp路徑下,有所有的訓練結果。
包括weights文件夾下有last.pt,best.pt。





