Toybrick

标题: ssd-mobilenet_v1转换失败 [打印本页]

作者: lijingle 时间: 2020-3-20 15:42
标题: ssd-mobilenet_v1转换失败
本帖最后由 lijingle 于 2020-3-20 15:45 编辑

例子是可以正常转换的。

我训练出来的模型进行转换
首先两种方式进行转换，第一种
如下：
from rknn.api import RKNN

INPUT_SIZE = 300

if __name__ == '__main__':
rknn = RKNN()
rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2')

print('--> Loading model')
rknn.load_tensorflow(tf_pb='frozen_inference_graph.pb',
                     inputs=['image_tensor'],
                     outputs=['detection_boxes','detection_scores'],
                     input_size_list=[[300,300,3]])
print('done')

print('--> Building model')
rknn.build(do_quantization=False)
print('done')

rknn.export_rknn('./digital_gesture.rknn')

rknn.release()

错误结果如下：
[attach]912[/attach]

好像是说有一层不支持

第二种

[attach]913[/attach]

直接在例子中替换了我训练的文件。
还是失败。

问下解决方法

作者: lijingle 时间: 2020-3-20 15:50
帖子和下面这个帖子错误基本相同
http://t.rock-chips.com/forum.ph ... &extra=page%3D2
但是没看到解决办法，

作者: leok 时间: 2020-3-21 16:14

lijingle 发表于 2020-3-20 15:50
帖子和下面这个帖子错误基本相同
http://t.rock-chips.com/forum.php?mod=viewthread&tid=1353&extra=page% ...

可以把模型发出来看看

作者: lijingle 时间: 2020-3-26 09:15
本帖最后由 lijingle 于 2020-3-26 10:39 编辑

模型位置
https://pan.baidu.com/s/1RlokT-YbZKOhYR2TZ_Xogg
提取码：e63q

作者: leok 时间: 2020-3-26 15:06

lijingle 发表于 2020-3-26 09:15
模型位置
https://pan.baidu.com/s/1RlokT-YbZKOhYR2TZ_Xogg
提取码：e63q

tf版本是多少？
输入大小是多少？

作者: leok 时间: 2020-3-26 15:16
rknn-toolkit 1.3.0 匹配1.10.0<tensorflow<1.13.2

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