Toybrick

Not surpport tensorflow version of yolov4-tiny

elexxy

新手上路

积分
45
楼主
发表于 2020-8-26 09:28:28    查看: 8838|回复: 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

版主

积分
13582
板凳
发表于 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


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