https://github.com/kaka-lin/object-detection
Object detection with ssd_mobilenet and tiny-yolo (Add: YOLOv3, tflite)
https://github.com/kaka-lin/object-detection
detection keras object-detection real-time-detection ssd-mobilenet tensorflow tiny-yolo video-detection yolov3
Last synced: 4 months ago
JSON representation
Object detection with ssd_mobilenet and tiny-yolo (Add: YOLOv3, tflite)
- Host: GitHub
- URL: https://github.com/kaka-lin/object-detection
- Owner: kaka-lin
- License: mit
- Created: 2018-02-14T08:36:27.000Z (over 7 years ago)
- Default Branch: master
- Last Pushed: 2020-03-05T09:02:58.000Z (over 5 years ago)
- Last Synced: 2024-12-31T13:03:27.335Z (5 months ago)
- Topics: detection, keras, object-detection, real-time-detection, ssd-mobilenet, tensorflow, tiny-yolo, video-detection, yolov3
- Language: Python
- Homepage:
- Size: 58.5 MB
- Stars: 190
- Watchers: 8
- Forks: 46
- Open Issues: 3
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Object Detection
* training:
- [yolov2-tf2](https://github.com/kaka-lin/yolov2-tf2)
- [yolov3-tf2](https://github.com/kaka-lin/yolov3-tf2)
* model (Inference):
- [tiny-YOLOv2](#tiny-yolo)
- [YOLOv3](#yolov3)
- [SSD-MobileNet v1](#ssd-mobilenet-v1)
- [SSDLite-MobileNet v2 (tflite)](#ssdlite-mobilenet-v2)
## Usage
### 1. tiny-YOLOv2
* download the [tiny-yolo](https://drive.google.com/file/d/14-5ZojD1HSgMKnv6_E3WUcBPxaVm52X2/view?usp=sharing) file and put it to model_data file
```baash
$ python3 test_tiny_yolo.py
```
### 2. YOLOv3
* download the [yolov3](https://drive.google.com/open?id=1vdD9TPiTWqvPxtCXdbVSKKksSdu0j_Hn) file and put it to model_data file
```baash
$ python3 test_yolov3.py
```
### 3. SSD-MobileNet v1
```baash
$ python3 test_ssd_mobilenet_v1.py
```
### 4. SSDLite-MobileNet v2 (tflite)
* download the [ssdlite-mobilenet-v2](https://drive.google.com/file/d/1Ha9yfjkweCatEo6UoZgZyHMeyIBGe5FO/view?usp=sharing) file and put it to model_data file
```baash
$ python3 test_ssdlite_mobilenet_v2.py
```
## Compare
* tiny-YOLOv2
* YOLOv3
* SSD-MobileNet v1
* SSDLite-MobileNet v2 (tflite)
## Acknowledgments
* Thanks to [keras-yolo3](https://github.com/qqwweee/keras-yolo3) for yolov3-keras part.
* Thanks to [mobile-object-detector-with-tensorflow-lite](https://medium.com/datadriveninvestor/mobile-object-detector-with-tensorflow-lite-9e2c278922d0) for ssdlite-mobilenet-v2 part.