TF2_YOLOv3

Tensorflow 2.1版的YOLOv3，參考許多前人的寫法並且修改一些個人感覺奇怪的部分。

權重轉換的converty.py由qqwweee 修改而來。

裡面包含一些詳細註解。

References:

qqwweee

zzh8829

YunYang1994

環境:

1. Tensorflow 2.1
2. Python 3.5~3.7
3. OpenCV 3~4

使用訓練好的weights

至YOLO 下載對應的權重，並且使用converty.py轉換為.h5檔案。

python convert.py yolov3.cfg yolov3.weights yolo.h5

python python test_img.py

則可看到下圖結果

training

需修改train.py。 使用labelImg 生成的xml檔即可。 如要使用YOLOv3提供的weights，類別數又不同時，在convert時，要先修改yolo.cfg相關層。