Toybrick

转换Tensorflow的pb模型问题?

zengwubin

中级会员

积分
274
楼主
发表于 2020-3-21 18:38:49    查看: 7575|回复: 1 | [复制链接]    打印 | 只看该作者
# Config for Model Input PreProcess
    rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2')

    # Load TensorFlow Model
    print('--> Loading model')
    rknn.load_tensorflow(tf_pb='./mtcnn_1.12.pb',
                         inputs=['input'],
                         outputs=['prob', 'landmarks', 'box'],
                         input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])
    print('done')

请问inputs和outputs该怎么写???
回复

使用道具 举报

jefferyzhang

版主

积分
13580
沙发
发表于 2020-3-21 20:27:10 | 只看该作者
论坛教程那么多都可以参考。
这个input output就是节点名字,你用Netron就可以看到。
如果你是tensorflow,你自己存的pb文件自己用tf读的时候也是要指定input output name的。
回复

使用道具 举报

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

本版积分规则

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


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