Toybrick

Not surpport tensorflow version of yolov4-tiny

elexxy

新手上路

积分
45
发表于 2020-8-26 09:28:28    查看: 5771|回复: 2 | [复制链接]    打印 | 显示全部楼层
本帖最后由 elexxy 于 2020-8-26 09:31 编辑

E Try match Split split_2 failed, catch exception!
W ----------------Warning(2)----------------
E Catch exception when loading tensorflow model: yolov4-tiny/yolov4-tiny.pb!
E Traceback (most recent call last):
E   File "rknn/base/RKNNlib/converter/convert_tf.py", line 713, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
E AttributeError: 'Split' object has no attribute 'load_params_from_tf'
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 374, in rknn.api.rknn_base.RKNNBase.load_tensorflow
E   File "rknn/base/RKNNlib/converter/convert_tf.py", line 767, in rknn.base.RKNNlib.converter.convert_tf.convert_tf.match_paragraph_and_param
E   File "rknn/api/rknn_log.py", line 312, in rknn.api.rknn_log.RKNNLog.e
E ValueError: Try match Split split_2 failed, catch exception!

For some reason I have to use one channel images as inputs, but the rknn toolkit does not support one channel darknet model's quantization.So I convert the darknet model to a tensorflow model and it works well before.However, today I try the yolov4 tiny model in this way, and the rknn toolkit return above error.It seems like dose not support the tf.split function.I hope you guys can support more functions and notice the problem of one channel darknet model's quantization.Tanks.


回复

使用道具 举报

elexxy

新手上路

积分
45
 楼主| 发表于 2020-8-26 09:34:16 | 显示全部楼层
PS. I have try it on 1.3.2 and 1.4.0 version, and it dose not work.
回复

使用道具 举报

jefferyzhang

版主

积分
12844
发表于 2020-8-26 14:44:17 | 显示全部楼层
NPU Team repeat:
We have successfully converted darknet offical edition yolov4-tiny (3-channels version) to rknn with rknn-toolkit 1.4.0.
Please try it. And your LOG looks can not support the op "split_2" in tf-edittion.
回复

使用道具 举报

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

本版积分规则

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


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