Toybrick

标题: 转换Tensorflow的pb模型问题？ [打印本页]

作者: zengwubin 时间: 2020-3-21 18:38
标题: 转换Tensorflow的pb模型问题？
# 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 时间: 2020-3-21 20:27
论坛教程那么多都可以参考。
这个input output就是节点名字，你用Netron就可以看到。
如果你是tensorflow，你自己存的pb文件自己用tf读的时候也是要指定input output name的。

欢迎光临 Toybrick (https://t.rock-chips.com/)