Object Detection

training:
- yolov2-tf2
- yolov3-tf2
model (Inference):

Usage

1. tiny-YOLOv2

download the tiny-yolo file and put it to model_data file

$ python3 test_tiny_yolo.py

2. YOLOv3

download the yolov3 file and put it to model_data file

$ python3 test_yolov3.py

3. SSD-MobileNet v1

$ python3 test_ssd_mobilenet_v1.py

4. SSDLite-MobileNet v2 (tflite)

download the ssdlite-mobilenet-v2 file and put it to model_data file

$ python3 test_ssdlite_mobilenet_v2.py

Compare

tiny-YOLOv2

YOLOv3

SSD-MobileNet v1

SSDLite-MobileNet v2 (tflite)

Acknowledgments

Thanks to keras-yolo3 for yolov3-keras part.
Thanks to mobile-object-detector-with-tensorflow-lite for ssdlite-mobilenet-v2 part.