object-detection-yolo

Introduction

This repository is an implementation of YOLOv3 in Tensorflow. It contains complete pipeline for training and prediction on custom datasets.
The primary references are yolo3-tf2 and keras-yolo3 which helped me to understand the implementations and bring up this repo.

Predict

convert pre-trained Darknet weights to h5 format

# yolov3
wget https://pjreddie.com/media/files/yolov3.weights -O data/yolov3.weights
python convert.py --weights_path data/yolov3.weights --output_path data/yolov3_coco.h5

detection on a single image

# yolov3
python predict.py -w data/yolov3_coco.h5 -c data/coco_classes.txt -i data/sample/apple.jpg

Training

pending to update

Benchmark / Result

pending to update

References

The various resources referred are organised and listed below.

YOLOv1

• practical-guide-object-detection-yolo-framework
• yolo-explained

YOLOv2

• yolo2-walkthrough-with-examples

model build - data loading - train - predict

• YOLO_Explained
• YAD2K

YOLOv3

• understanding-yolo-and-implementing-yolov3-for-object-detection
• all-you-need-to-know-about-yolo-v3-you-only-look-once
• a-comprehensive-guide-to-yolov3
• yolo-you-look-only-once
• yolo-v3-explained
• yolo-v3-object-detection
• bounding-box-object-detectors-understanding-yolo
• How YOLO V3 works?
• YOLOv3 from Scratch
• Introduction into YOLO v3

Implementation in pyTorch

model build - data loading - train - predict

• implement-yolo-v3-object-detector-pytorch
• yolov3-implementation-with-training-setup
• pytorch/object_detection/YOLOv3

Implementation in Tensorflow

pre trained model - predict

• yolov3-tensorflow
• implement_yolo

model build - data loading - train - predict

• yolo3-tf2
• yolo3-tf2 (good details for concepts - model/loss/bbox)
• keras-yolo3 (good class organisation, generator, and augmentation)
• qqwweee-keras-yolo3
• TensorFlow-2.x-YOLOv3
• YOLOv3-object-detection-custom-training
• YOLOv3-TF2-introduction
• yolov3-tf2
• tensorflow-yolo-v3
• keras-YOLOv3-model

custom training

• training-yolov3-with-a-custom-dataset

transfer learning with yolo

• types-of-deep-transfer-learning
• transfer-learning-with-yolo-v3
• yolov3-custom-object-detection-with-transfer-learning

load yolo weights

• darknet-yolo
• tensorflow-yolo-v3
• yolov3-tensorflow
• YOLOv3-custom-training
• implementing-yolo-v3-in-tensorflow
• how-to-perform-object-detection-with-yolov3
• yolo3_one_file_to_detect_them_all

YOLOv4

load yolo weights

• how-to-convert-your-yolov4-weights
• how-to-convert-yolov4-from-weights-to-h5
• yolov4_to_tf2

YOLOv5

• how-to-use-yolo-v5-object-detection

Model Subclassing

• model-sub-classing

Custom Loss

• keras-custom-loss-functions
• kaggle-discussion

Dataset / Format

• coco-data-format-for-object-detection
• cocodataset
• pascal VOC
• 2D Bounding box annotation formats for Object detection

Todo

• Update the training and benchmark / result sections