iOS-CoreML-Yolo在CoreML框架上使用Tiny YOLO v1模型实现对象检测

iOS-CoreML-Yolo在CoreML框架上使用Tiny YOLO v1模型实现对象检测

iOS-CoreML-Yolo

This is the implementation of Object Detection using Tiny YOLO v1 model on Apple's CoreML Framework.

The app fetches image from your camera and perform object detection @ (average) 17.8 FPS.

Requirements

Xcode 9 and aboveiOS 11 and aboveFor training: Python 2.7 (Keras 1.2.2, TensorFlow 1.1, CoreMLTools 0.7)

Usage

To use this app, open iOS-CoreML-MNIST.xcodeproj in Xcode 9 and run it on a device with iOS 11. (You can also use simulator)

Model conversion

In this project, I am not training YOLO from scratch but converting the already existing model to CoreML model. If you want to create model on your own.

Create Anaconda environment. Open your terminal and type the following commands.

$ conda create -n coreml python=2.7$ source activate coreml(coreml) $ conda install pandas matplotlib jupyter notebook scipy scikit-learn opencv(coreml) $ pip install tensorflow==1.1(coreml) $ pip install keras==1.2.2(coreml) $ pip install h5py(coreml) $ pip install coremltools

Download the weights from the following link and move it into ./nnet directory.Enter the environment and run the following commands in terminal with ./nnet as master directory.

(coreml) $ sudo python convert.py

I also included a jupyter notebook for better understanding the above code. You need to use it with root permissions for mainly converting the keras model to CoreML model. Initialise the jupyter notebook instance with the following command:

(coreml) $ jupyter notebook --allow-root

CoreML Model Download [New]

The converted CoreML model can be downloaded here:

Google Drive Link: Download

Tutorial

If you are interested in creating the Tiny YOLO v1 model on your own, a step-by-step tutorial is available at - Link

Results

These are the results of the app when tested on iPhone 7.

Author

Sri Raghu Malireddi / @r4ghu

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。