gplast / tflite-yolov3-gpu-ready

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

2. gonogono - Observing traffic signs and front car instead of the driver

Please refer to iOS/gonogono directory of this repository.

Future update

  • Make model smaller -> under 5MB
  • Int8 quantization support
  • Android example application

About

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


Languages

Language:Swift 63.6%Language:Python 35.4%Language:Shell 0.7%Language:Ruby 0.3%