/object-detection

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

Primary LanguagePythonMIT LicenseMIT

Object Detection

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)

$ python3 test_ssdlite_mobilenet_v2.py 

Compare

  • tiny-YOLOv2

  • YOLOv3

  • SSD-MobileNet v1

  • SSDLite-MobileNet v2 (tflite)

Acknowledgments