索引
一、安装NCSDK
地址:https://github.com/movidius/ncsdk
安装步骤就跟着官方来,很简单
在make examples时,我出现了2代棒USB无法识别的问题:[Error 7] Toolkit Error: USB Failure. Code: Error opening devic
原因:英特尔说2代不支持NCSDK,大家得去用它新出的OpenVINO的SDK(REFRENCE)
二、NCSDK的几个指令
其实在/ncsdk/examples/中的makefile都有说明,在此总结
在编译自己的模型时,我们重点关注一下两个指令(其实也是一个)
1、mvNCCompile
这个是编译自己的模型并生成能够被神经棒识别的graph的重要指令
使用方法:
mvNCCompile model/path -in=input_node_name -on=output_node_name
重要的就是这三个参数,具体可以mvNCCompile -h查看,但是有些限制
(0)本身这个神经棒能做的运算就有局限,例如MobileNet就没办法跑因为有不支持的层
会抛错[Error 5] Toolkit Error: Stage Details Not Supported: IsVariableInitialized
(1)必须是单个输入和单个输出(可能3个也可以,还没试验,后期补充)
(2)必须是整个模型完整的输入输出,不能是中间层
(3)模型大小有限制,我测试VGG16模型>500M抛出
/usr/local/bin/ncsdk/Controllers/FileIO.py:65: UserWarning: You are using a large type. Consider reducing your data sizes for best performance
并且影响后续的mvNCProfile抛错
2、mvNCProfile
是在complie的基础上生成评估的结果,和Compile用法相同
3、mvNCCheck
对比在CPU/GPU和神经棒上测试的结果对比,包括时长、accuracy等
三、输入输出Node名字
我是基于keras(tf后端)做的,一开始不知道命名方式,总是不能正确的给on赋值
目前我的解决方式是在保存模型的时候加一行
conf = sess.graph.get_operations()
print(conf)
然后把结果打印出来,然后再找输出节点的名字
但我总结规律,大概率是
output-layer-name/Operation-name
例如predictions/Softmax,reshape_2/Reshape
欢迎各位指正啊!