yolov3-ios

Using yolo v3 object detection on ios platform.

Example applications:

QuickStart:

Run tiny_model.xcodeproj in ios.

Training

The training process mainly consults qqwweee/keras-yolo3. We add yolov3 with Densnet.

1.Requirement

python 3.6.4
keras 2.1.5
tensorflow 1.6.0

2.Generate datasets

Generate datasets with VOC format. And try python voc_annotations.

3.Start training

cd yolov3_with_Densenet

For yolo model with darknet:

wget https://pjreddie.com/media/files/darknet53.conv.74
rename it as darknet53.weights
python convert.py -w darknet53.cfg darknet53.weights model_data/darknet53_weights.h5
python yolov3_train.py, with model_data/darknet53_weights.h5 as pre-trained model

For yolo model with densenet：

python densenet_train.py, with model_data/dense121_weights.h5 as pre-trained model

Converting

1.Building environment

virtualenv -p /usr/bin/python2.7 keras_coreml_virt
source keras_coreml_virt/bin/activate
pip install protobuf
pip install tensorflow==1.6.0
pip install keras==2.1.5
pip install h5py
pip install coremltools==0.8.0

2.Convert .h5 model to .mlmodel

python coreml.py

Building project in Xcode

open tiny_model.xcodeproj with Xcode 9+
change the .mlmodel file and Target Menmbership

For yolo model with darknet or densenet

modify the code from line 43 to line 49 in YOLO.swift
change the labels and the anchors in Helpers.swift
run the project

For tiny model

just change the labels and run the project

guidefreitas / yolov3-ios