kaka-lin/object-detection

Object detection with ssd_mobilenet and tiny-yolo (Add: YOLOv3, tflite)

Object Detection

• training:

  • yolov2-tf2
  • yolov3-tf2

• model (Inference):

  • tiny-YOLOv2
  • YOLOv3
  • SSD-MobileNet v1
  • SSDLite-MobileNet v2 (tflite)

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.