THIS REPO IS CUSTOMISED FROM https://github.com/pythonlessons/TensorFlow-2.x-YOLOv3
YOLOv3 implementation in TensorFlow 2.x, with support for training, transfer learning.
First, clone or download this GitHub repository. Install requirements and download pretrained weights:
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
Custom training required to prepare dataset first, how to prepare dataset and train custom model you can read in following link:
https://pylessons.com/YOLOv3-TF2-custrom-train/