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

程式碼

前陣子在做華為垃圾偵測專案的時候,就遇到了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提供的程式除了做格式轉換外還能按照比例切割訓練 / 測試 /驗證集,省下自己寫切割資料集程式的時間。

Machine Learning | Deep Learning | https://linktr.ee/yanwei