Toybrick

ssd-mobilenet_v1转换失败

lijingle

注册会员

积分
101
楼主
发表于 2020-3-20 15:42:47    查看: 5748|回复: 5 | [复制链接]    打印 | 只看该作者
本帖最后由 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()



错误结果如下:


好像是说有一层不支持

第二种



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

问下解决方法




本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

lijingle

注册会员

积分
101
沙发
 楼主| 发表于 2020-3-20 15:50:41 | 只看该作者
帖子和下面这个帖子错误基本相同
http://t.rock-chips.com/forum.ph ... &extra=page%3D2
但是没看到解决办法,
回复

使用道具 举报

leok

版主

积分
894
板凳
发表于 2020-3-21 16:14:50 | 只看该作者
lijingle 发表于 2020-3-20 15:50
帖子和下面这个帖子错误基本相同
http://t.rock-chips.com/forum.php?mod=viewthread&tid=1353&extra=page% ...

可以把模型发出来看看
回复

使用道具 举报

lijingle

注册会员

积分
101
地板
 楼主| 发表于 2020-3-26 09:15:32 | 只看该作者
本帖最后由 lijingle 于 2020-3-26 10:39 编辑

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

使用道具 举报

leok

版主

积分
894
5#
发表于 2020-3-26 15:06:31 | 只看该作者
lijingle 发表于 2020-3-26 09:15
模型位置
https://pan.baidu.com/s/1RlokT-YbZKOhYR2TZ_Xogg
提取码:e63q

tf版本是多少?
输入大小是多少?
回复

使用道具 举报

leok

版主

积分
894
6#
发表于 2020-3-26 15:16:11 | 只看该作者
rknn-toolkit 1.3.0 匹配1.10.0<tensorflow<1.13.2
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表