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

程式碼

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

介紹

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

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store