Toybrick

标题: tensorflow训练inception v3模型转rknn，config怎么改 [打印本页]

作者: yangting0501 时间: 2019-11-15 10:44
标题: tensorflow训练inception v3模型转rknn，config怎么改
tensorflow训练出的inception v3图像分类模型转rknn时，由于inception v3模型训练时图片进行了如下操作，image = tf.image.convert_image_dtype(image, dtype=tf.float32)，所以图像的像素值范围是(0,1)之间，所以我将config设置为rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2')。但这样转出来的rknn模型跑出来的结果与tf模型差异很大。请问rknn.config里的参数应该怎么修改。

作者: yangting0501 时间: 2019-11-18 14:03
训练分类模型时，将输入图像处理的过程改成了每个通道减128，然后除以128。转rknn时rknn.config(channel_mean_value='128 128 128 128', reorder_channel='0 1 2')。

作者: yangting0501 时间: 2019-11-18 14:25
本帖最后由 yangting0501 于 2019-11-18 14:27 编辑

转模型时关闭了量化，输出的分类值还是不一样。
训练模型时图像处理方式：
C:\Users\Ting0501\Desktop\微信截图_20191118142250.png

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