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 (http://t.rock-chips.com/) Powered by Discuz! X3.3