Python影像辨識筆記(二十五)：將VOC2007格式的資料集透過PyTorch版本的YOLOv3進行訓練

程式碼

pprp/voc2007_for_yolo_torch

前陣子在做華為垃圾偵測專案的時候，就遇到了voc格式的轉換問題，當時是使用convert2Yolo搭配以下指令進行轉換：

python3 example.py --datasets VOC --img_path ../trainval/VOC2007/JPEGImages --label ../trainval/VOC2007/Annotations/ --convert_output_path ./yolo/ --img_type ".jpg" --manifest_path ./ --cls_list_file ../trainval/train_classes.txt

介紹

但是在pprp/voc2007_for_yolo_torch當中，作者提供了另一種進行格式轉換的方式。與convert2Yolo兩者都是不錯的選擇，但pprp提供的程式除了做格式轉換外還能按照比例切割訓練 / 測試 /驗證集，省下自己寫切割資料集程式的時間。