YOLO v3 TensorFlow Lite iOS GPU acceleration
August 14, 2020 ยท View on GitHub
- This projects trains YOLO v3 model with BDD100k Dataset
- And you can convert the trained model into TFLite model that GPU acceleration ready for iOS!
Training
Please refer to python directory of this repository.
iOS Examples
You can find out iOS directory of this repository.
1. YOLO object detection example project
Please refer to iOS/yolojk_iOS directory of this repository.
| iPhone X | iPadPro 11" |
|---|---|
 |  |
2. gonogono - Observing traffic signs and front car instead of the driver
Please refer to iOS/gonogono directory of this repository.
Performance
| Model Name | # Params | # Label | Device | FPS |
|---|---|---|---|---|
| MobileNetV1 | ~9M | 4 | iPhone X | 15~20 |
| MobileNetV1 | ~2.3M | 4 | iPhone X | 30~35 |
| MobileNetV1 | ~2.3M | 4 | iPad Pro 11" | 60~70 |
Future update
Make model smaller -> under 5MB- Int8 quantization support
- Android example application
- CoreML Support