/tflite-yolov3-gpu-ready

YOLO v3 TensorFlow Lite iOS GPU acceleration. Training from scratch and making a GPU accelerated mobile application.

Primary LanguageSwift

YOLO v3 TensorFlow Lite iOS GPU acceleration

  • 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