Toybrick

RKNN 转换 mobile V1 ssd 报错,请问 这个是为什么?已经使用.

yhc

注册会员

积分
177
楼主
发表于 2019-2-18 12:05:13 | 显示全部楼层
确认是ssd_mobilenet_v1?
如果是ssd_mobilenet_v2的input节点有点不一样
  1.     rknn.load_tensorflow(tf_pb='./ssd_mobilenet_v2_coco_2018_03_29/frozen_inference_graph.pb',
  2.                          inputs=['FeatureExtractor/MobilenetV2/MobilenetV2/input'],
  3.                          outputs=['concat', 'concat_1'],
  4.                          input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])
复制代码
回复

使用道具 举报

yhc

注册会员

积分
177
沙发
发表于 2019-2-18 15:53:48 | 显示全部楼层
可以用netron打开pb模型看网络结构,取这个节点是有去掉了一些前处理
回复

使用道具 举报

yhc

注册会员

积分
177
板凳
发表于 2019-2-19 16:24:24 | 显示全部楼层
一般的话要自己熟悉网络的结构,就ssd_mobilenet_v2网络来说,前处理的话都是Preprocessor开头的,所以就跳过这些Preprocessor的节点,取到
FeatureExtractor/MobilenetV2/MobilenetV2/input,因为这些前处理节点如果用NPU来跑并不是效率最高的,可以用cpu来替代处理
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表