Toybrick

标题: tf-yolov3-tiny模型转化出错 [打印本页]

作者: christian 时间: 2020-3-19 09:37
标题: tf-yolov3-tiny模型转化出错
在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

作者: jefferyzhang 时间: 2020-3-19 09:51
你这已经转换完了，跑模型出的错。
这个你要自己看下用的仿真环境还是计算棒，还是什么开发版，固件环境有没有问题，驱动版本是不是匹配

作者: christian 时间: 2020-3-19 10:08

jefferyzhang 发表于 2020-3-19 09:51
你这已经转换完了，跑模型出的错。
这个你要自己看下用的仿真环境还是计算棒，还是什么开发版，固件环境有 ...

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

作者: jefferyzhang 时间: 2020-3-19 10:24

christian 发表于 2020-3-19 10:08
用的是PC的仿真环境，用的是rknn-toolkit-V1.3.0，用官方提供的ssd的demo跑是没问题的。这个跟输入节点会 ...

应该跟你配置有关系。
yolo3 tiny我们是有教程的，没问题的

作者: christian 时间: 2020-3-19 10:33

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大小有关的？

作者: jefferyzhang 时间: 2020-3-19 10:39
你这个tf模型自己能导入用么？TF版本只能用1.13
1. Warning不影响
2. 你tf模型是不是有问题

作者: christian 时间: 2020-3-19 10:46

jefferyzhang 发表于 2020-3-19 10:39
你这个tf模型自己能导入用么？TF版本只能用1.13
1. Warning不影响
2. 你tf模型是不是有问题 ...

TF导入是可以用的，TF的1.13.1版本也不支持吗？我的TF版本是1.13.1的

作者: jefferyzhang 时间: 2020-3-19 10:49

christian 发表于 2020-3-19 10:46
TF导入是可以用的，TF的1.13.1版本也不支持吗？我的TF版本是1.13.1的

1.13.x都可以。
你看过我们教程么，我们教程就是yolov3 tiny

作者: christian 时间: 2020-3-19 10:51

jefferyzhang 发表于 2020-3-19 10:49
1.13.x都可以。
你看过我们教程么，我们教程就是yolov3 tiny

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

作者: jefferyzhang 时间: 2020-3-19 10:55

christian 发表于 2020-3-19 10:51
记得教程里用的是darknet的，我用darknet训练出来的模型精度不行，TF训练出来的模型精度还可以的。 ...

嗯？你的精度是如何计算可以不可以的？
你看我们教程，也是tf训练的，然后转回darknet转rknn的。

作者: christian 时间: 2020-3-19 11:08

jefferyzhang 发表于 2020-3-19 10:55
嗯？你的精度是如何计算可以不可以的？
你看我们教程，也是tf训练的，然后转回darknet转rknn的。
...

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

作者: christian 时间: 2020-3-19 11:11

jefferyzhang 发表于 2020-3-19 10:55
嗯？你的精度是如何计算可以不可以的？
你看我们教程，也是tf训练的，然后转回darknet转rknn的。
...

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

作者: jefferyzhang 时间: 2020-3-19 11:11

christian 发表于 2020-3-19 11:11
是在测试集上测试，tf要优于darknet，具体原因还没分析。教程中没有看到tf转darknet的过程，能否帮忙提供 ...

教程都有源码的，请仔细阅读源码和readme，都开源的。

作者: christian 时间: 2020-3-19 11:28

jefferyzhang 发表于 2020-3-19 11:11
教程都有源码的，请仔细阅读源码和readme，都开源的。

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

作者: jefferyzhang 时间: 2020-3-19 11:35

christian 发表于 2020-3-19 11:28
在rknn-toolkit-V1.3.0下的examples中木有看到相关源码，doc中也木有看到相关的说明，能否提供下链接的， ...

看置顶贴整合。
教程不会放在rknntoolkit里的

作者: christian 时间: 2020-3-19 13:15

jefferyzhang 发表于 2020-3-19 11:35
看置顶贴整合。
教程不会放在rknntoolkit里的

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

作者: christian 时间: 2020-3-19 15:50

hisping 发表于 2020-3-19 14:52
相同的模型相同的数据集不存在说tf训练出来的比darknet训练的效果好，应该是差不多的，除非你训练的有问 ...

好的，谢谢，我试下看。

作者: ofqinqin 时间: 2020-9-25 18:20

christian 发表于 2020-3-19 15:50
好的，谢谢，我试下看。

我这边也是同样的问题tensorflow版本的yolov3-tiny，pb模型没有问题，转化成功，调用失败，请帮忙看看
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.
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

作者: ofqinqin 时间: 2020-9-25 18:22

jefferyzhang 发表于 2020-3-19 11:35
看置顶贴整合。
教程不会放在rknntoolkit里的

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