|
我用 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]])
|
|