Toybrick

标题: deeplab tensorflow 模型转换问题 [打印本页]

作者: raymond 时间: 2019-7-2 15:53
标题: deeplab tensorflow 模型转换问题
我用 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]])

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