Tensorflow 2.1版的YOLOv3,參考許多前人的寫法並且修改一些個人感覺奇怪的部分。
權重轉換的converty.py由qqwweee 修改而來。
裡面包含一些詳細註解。
- Tensorflow 2.1
- Python 3.5~3.7
- OpenCV 3~4
至YOLO 下載對應的權重,並且使用converty.py轉換為.h5檔案。
python convert.py yolov3.cfg yolov3.weights yolo.h5
python python test_img.py
則可看到下圖結果
需修改train.py。 使用labelImg 生成的xml檔即可。 如要使用YOLOv3提供的weights,類別數又不同時,在convert時,要先修改yolo.cfg相關層。