YOLOv3-TensorFlow-2.x

YOLOv3 implementation in TensorFlow 2 version. My Blog post https://medium.com/analytics-vidhya/yolov3-object-detection-in-tensorflow-2-x-8a1a104c46a8

Installation

First, clone or download this GitHub repository. Install requirements and download pretrained weights:

git clone https://github.com/anushkadhiman/YOLOv3-TensorFlow-2.x.git
cd YOLOv3-TensorFlow-2.x

pip install -r ./requirements.txt

# yolov3
wget -P model_data https://pjreddie.com/media/files/yolov3.weights

# yolov3-tiny
wget -P model_data https://pjreddie.com/media/files/yolov3-tiny.weights

Detection

Start with using pretrained weights to test predictions on both image and video:

python detection_demo.py

Result

Detection by Pre-trained model

Detection by Custom Trained model

References

1. https://github.com/pjreddie/darknet

• Official YOLOv3 implementation

2. https://github.com/AlexeyAB

• Explanations of parameters

3. https://github.com/pythonlessons/TensorFlow-2.x-YOLOv3

• Models, loss functions

4. https://pylessons.com/ — Rokas Balsys