Movidius VPU移植ssd_mobilenet问题记录

ssd_mobilenet移植到Movidius到目前为止我还没移植成功,这里记录一下遇到的一些问题,以及后续的研究方向。

首先,我们在 detection_models_zoo中把ssdlite_mobilenet_v2_coco算法模型下载下来,下载后的tar包中有如下__图二__中的几个文件:

这里写图片描述
checkpoint:这个文件是检查点文件,该文件是在网络训练过程中通过tf.train.Saver保存的当前批次训练结束后的检查点文件。
frozen_inference_graph.pb:这个文件是生成的一个可以用来导入到tflite中的计算图,我们在tensorflow中调用该模型进行目标检测时正是通过解析该文件来获取计算图和参数集。
model.ckpt.*:这三个文件和前面的__checkpoint__文件是对应出现的,model.ckpt.data*是所有weights和bias的值,model.ckpt.meta是计算图,model.ckpt.index是索引文件。
pipeline.config:是保存一些模型入参的配置文件,是用来我们训练自己的模型时进行调参用。

所以按照ncsdk/docs/tf_compile_guidance.html描述,再结合上面这几个文件作用的分析,我使用的编译命令如下:

mvNCCompile frozen_inference_graph.pb -s 12 -in input -on output -o ssd_mobilenet.graph
mvNCCompile:Ncsdk的编译工具,用来生成计算图。
-s:  使用SHAVE的个数。
-in: 网络模型中输入层node的名字。
-on: 网络模型中输出层node的名字。
-o:  编译后生成的文件名。

运行之后,问题来了:
问题一:input node找不到

$ mvNCCompile frozen_inference_graph.pb -s 12 -in input -on output -o ssd_mobilenet.graph
mvNCCompile v02.00, Copyright @ Intel Corporation 2017

/usr/local/lib/python3.5/dist-packages/tensorflow/python/util/tf_inspect.py:45: DeprecationWarning: inspect.getargspec() is deprecated, use inspect.signature() instead
Traceback (most recent call last):
  File "/usr/local/bin/mvNCCompile", line 169, in <module
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 17
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 17
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值