Toybrick

caffe 模型转换问题

guigen80

新手上路

积分
31
发表于 2019-3-21 17:03:00    查看: 8078|回复: 2 | [复制链接]    打印 | 显示全部楼层
python3 rknn_transform_caffe.py
--> Loading model
E Catch exception when loading caffe model: ./yolo_relu/yolov2_caffe_relu_0_merge_int8.caffemodel!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 246, in rknn.api.rknn_base.RKNNBase.load_caffe
T   File "rknn/base/RKNNlib/converter/caffeloader.py", line 958, in rknn.base.RKNNlib.converter.caffeloader.CaffeLoader.load
T   File "rknn/base/RKNNlib/converter/caffeloader.py", line 945, in rknn.base.RKNNlib.converter.caffeloader.CaffeLoader.init_caffe_net
T   File "rknn/base/RKNNlib/converter/caffeloader.py", line 932, in rknn.base.RKNNlib.converter.caffeloader.CaffeLoader.import_caffe_proto
T   File "/usr/lib/python3.5/importlib/__init__.py", line 121, in import_module
T     raise TypeError(msg.format(name))
T TypeError: the 'package' argument is required to perform a relative import for './yolo_relu/gnet_region_deploy_darknet448_voc.prototxt_pb2'
done
--> Building model
E Catch exception when building RKNN model!
T Traceback (most recent call last):
T   File "rknn/api/rknn_base.py", line 459, in rknn.api.rknn_base.RKNNBase.build
T   File "rknn/api/rknn_base.py", line 939, in rknn.api.rknn_base.RKNNBase._quantize
T   File "rknn/base/RKNNlib/console/utils.py", line 7, in rknn.base.RKNNlib.console.utils.get_model_data_file
T   File "/usr/lib/python3.5/posixpath.py", line 122, in splitext
T     return genericpath._splitext(p, sep, None, extsep)
T   File "/usr/lib/python3.5/genericpath.py", line 118, in _splitext
T     sepIndex = p.rfind(sep)
T AttributeError: 'NoneType' object has no attribute 'rfind'
done
E RKNN model data is None, please load model first!



code:
rknn.load_caffe(model="./yolo_relu/yolov2_caffe_relu_0_merge_int8.caffemodel",proto="./yolo_relu/gnet_region_deploy_darknet448_voc.prototxt",blobs="data")




回复

使用道具 举报

guigen80

新手上路

积分
31
 楼主| 发表于 2019-3-22 09:59:25 | 显示全部楼层
有人能帮我看看怎么回事吗?谢谢
回复

使用道具 举报

guigen80

新手上路

积分
31
 楼主| 发表于 2019-3-22 17:50:44 | 显示全部楼层
问题解决了,是我参数输错了,不过我发现它还是不支持reorg层,yolov2无法完成转换
回复

使用道具 举报

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

本版积分规则

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


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