TensorFlow 2.0 YOLO-V3

這個專案實現了TensorFlow2.0 YOLO-V3物件偵測。

Usage

• Requirements

  • Numpy == 1.19.5

  • OpenCV == 4.5.2.54

  • Matplotlib == 3.1.1

  • TensorFlow == 2.5.0

  • TensorFlow Datasets == 4.3.0

  • Tensorflow Addons == 0.13.0

• DarkNet weights 轉換到 TensorFlow weights

  python convert.py

• YOLO-V3訓練

  python train.py

• YOLO-V3多尺度訓練(multi scale training)

  python train-multi-scale.py

• YOLO-V3測試

  python test.py

• TensorBoard觀察訓練紀錄

  tensorboard --logdir logs_yolo

Results

Predict by YOLO v3 original weights

References

• https://github.com/pjreddie/darknet
  • YOLO-V3原作者程式碼(DarkNet)
• https://github.com/qqwweee/keras-yolo3
  • 網路模型
• https://github.com/zzh8829/yolov3-tf2
  • 損失函數
  • 資料預處理
• https://github.com/allanzelener/YAD2K
  • DarkNet到Keras網路模型轉換