Toybrick

tf-yolov3-tiny模型转化出错

christian

注册会员

积分
132
楼主
发表于 2020-3-19 09:37:03    查看: 19434|回复: 18 | [复制链接]    打印 | 显示全部楼层
在PC ubuntu16.04 安装rknn-toolkit-V1.3.0,跑tensorflow的ssd_mobilenet_v1 demo是正常的,但是在tf-yolov3-tiny转化模型过程中出现如下错误,还请帮忙分析下:

  warnings.warn(message)
W Do not support shape > 4.
W Do not support shape > 4.
done
--> Building model
W Do not support shape > 4.
W Do not support shape > 4.
W Do not support shape > 4.
W Do not support shape > 4.
W Do not support shape > 4.
W Do not support shape > 4.
done
--> Init runtime environment
E [op_check:103]Invalid broadcast for inputs[0] size[3]
E [setup_node:383]Check node[52] ADD fail
Create RKNN model fail!
E Catch exception when init runtime!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 988, in rknn.api.rknn_base.RKNNBase.init_runtime
E   File "rknn/api/rknn_runtime.py", line 320, in rknn.api.rknn_runtime.RKNNRuntime.build_graph
E Exception: RKNN init failed. error code: RKNN_ERR_MODEL_INVALID
Init runtime environment failed



回复

使用道具 举报

christian

注册会员

积分
132
沙发
 楼主| 发表于 2020-3-19 10:08:11 | 显示全部楼层
jefferyzhang 发表于 2020-3-19 09:51
你这已经转换完了,跑模型出的错。
这个你要自己看下用的仿真环境还是计算棒,还是什么开发版,固件环境有 ...

用的是PC的仿真环境,用的是rknn-toolkit-V1.3.0,用官方提供的ssd的demo跑是没问题的。这个跟输入节点会有关系吗?
回复

使用道具 举报

christian

注册会员

积分
132
板凳
 楼主| 发表于 2020-3-19 10:33:06 | 显示全部楼层
jefferyzhang 发表于 2020-3-19 10:24
应该跟你配置有关系。
yolo3 tiny我们是有教程的,没问题的

我的是在Tensorflow训练的模型转的。日志里边有几个问题,还请帮忙看下。
1.警告W Do not support shape > 4. 会有影响吗?
2.[op_check:103]Invalid broadcast for inputs[0] size[3],这个是不是跟输入的shape大小有关的?
回复

使用道具 举报

christian

注册会员

积分
132
地板
 楼主| 发表于 2020-3-19 10:46:19 | 显示全部楼层
jefferyzhang 发表于 2020-3-19 10:39
你这个tf模型自己能导入用么?TF版本只能用1.13
1. Warning不影响
2. 你tf模型是不是有问题 ...

TF导入是可以用的,TF的1.13.1版本也不支持吗?我的TF版本是1.13.1的
回复

使用道具 举报

christian

注册会员

积分
132
5#
 楼主| 发表于 2020-3-19 10:51:54 | 显示全部楼层
jefferyzhang 发表于 2020-3-19 10:49
1.13.x都可以。
你看过我们教程么,我们教程就是yolov3 tiny

记得教程里用的是darknet的,我用darknet训练出来的模型精度不行,TF训练出来的模型精度还可以的。
回复

使用道具 举报

christian

注册会员

积分
132
6#
 楼主| 发表于 2020-3-19 11:08:16 | 显示全部楼层
jefferyzhang 发表于 2020-3-19 10:55
嗯?你的精度是如何计算可以不可以的?
你看我们教程,也是tf训练的,然后转回darknet转rknn的。
...

我的是自己训练出来的在测试集上测试看结果TF是要好点,具体原因还没研究。没有找到教程中tf转darknet的过程的?麻烦帮忙提供下吧,谢谢。
回复

使用道具 举报

christian

注册会员

积分
132
7#
 楼主| 发表于 2020-3-19 11:11:06 | 显示全部楼层
jefferyzhang 发表于 2020-3-19 10:55
嗯?你的精度是如何计算可以不可以的?
你看我们教程,也是tf训练的,然后转回darknet转rknn的。
...

是在测试集上测试,tf要优于darknet,具体原因还没分析。教程中没有看到tf转darknet的过程,能否帮忙提供下?谢谢。
回复

使用道具 举报

christian

注册会员

积分
132
8#
 楼主| 发表于 2020-3-19 11:28:40 | 显示全部楼层
jefferyzhang 发表于 2020-3-19 11:11
教程都有源码的,请仔细阅读源码和readme,都开源的。

在rknn-toolkit-V1.3.0下的examples中木有看到相关源码,doc中也木有看到相关的说明,能否提供下链接的,麻烦了,谢谢
回复

使用道具 举报

christian

注册会员

积分
132
9#
 楼主| 发表于 2020-3-19 13:15:55 | 显示全部楼层
jefferyzhang 发表于 2020-3-19 11:35
看置顶贴 整合。
教程不会放在rknntoolkit里的

版主,看了置顶帖还没找到。我看里边的yolov3是用darknet训练的,tensorflow的PD文件都是通过rknn.load_tensorflow转化的
回复

使用道具 举报

christian

注册会员

积分
132
10#
 楼主| 发表于 2020-3-19 15:50:29 | 显示全部楼层
hisping 发表于 2020-3-19 14:52
相同的模型相同的数据集不存在说tf训练出来的比darknet训练的效果好,应该是差不多的,除非你训练的有问 ...

好的,谢谢,我试下看。
回复

使用道具 举报

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

本版积分规则

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


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