Skip to content

Latest commit

 

History

History
43 lines (27 loc) · 1.12 KB

README.md

File metadata and controls

43 lines (27 loc) · 1.12 KB

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相關層。