Toybrick

deeplab tensorflow 模型转换问题

raymond

新手上路

积分
41
楼主
发表于 2019-7-2 15:53:54    查看: 7977|回复: 0 | [复制链接]    打印 | 只看该作者
我用 tensorflow 的 deeplabv3_mnv2_dm05_pascal_trainaug 模型转换成 rknn模型,模型转换是成功的,但是在3399pro上推测出来的输出层 logits/semantic/BiasAdd 结果完全不对我不知道是不是转换的问题,我在 server 上用同样的代码 rknn 模拟出来的结果和开发板上的结果也不一样


我转换的代码如下 (INPUT_SIZE = 513):
11     rknn.config(channel_mean_value='0. 0. 0. 1.', reorder_channel='0 1 2')
12
13     print('--> Loading model')
14     rknn.load_tensorflow(tf_pb=model_path,
15                          inputs=['MobilenetV2/Conv/Conv2D'],
16                          outputs=['logits/semantic/BiasAdd'],
17                          input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])




回复

使用道具 举报

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

本版积分规则

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


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