本文讨论在Ubuntu16.04的系统中使用Movidius神经计算棒,进行vgg16算法预测。
神经计算SDK官方网站https://movidius.github.io/ncsdk/install.html
1.首先在系统中安装python3.5,这里不再赘述。
2.然后进行神经计算SDK的安装,注意安装过程中记得将计算棒插入USB接口。
本文以NCSDK2的版本进行讲解,依次执行以下安装命令:
wget https://ncs-forum-uploads.s3.amazonaws.com/ncsdk/ncsdk-02_05_00_02-full/ncsdk-2.05.00.02.tar.gz
tar xvf ncsdk-2.05.00.02.tar.gz
cd ncsdk-2.05.00.02
make install
make examples
如果因为环境问题导致安装不成功,具体冲突可查看官网说明。
3.安装成功后,将caffe或者tensorflow训练好的模型进行编译,NCSDK会自动安装tensorflow1.7.0的版本。本文以vgg16为例,建立model_build.py:
from keras.applications import VGG16
from keras import backend as K
import tensorflow as tf
mn = VGG16()
saver = tf.train.Saver()
sess = K.get_session()
saver.save(sess, "./TF_Model/vgg16")
4.然后