Toybrick

楼主: hisping

人工智能开发系列(3) YOLOV3开发与实现

 

呱唧

注册会员

积分
59
楼主
发表于 2020-4-1 14:13:18 | 显示全部楼层
转换tiny的时候会报以下错误

$ python rknn_transform_tiny.py
--> Loading model
E Catch exception when loading darknet model: ./yolov3-tiny.cfg!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 511, in rknn.api.rknn_base.RKNNBase.load_darknet
E   File "rknn/base/RKNNlib/converter/convert_darknet.py", line 685, in rknn.base.RKNNlib.converter.convert_darknet.Converter.load_blobs
E   File "rknn/base/RKNNlib/converter/convert_darknet.py", line 403, in rknn.base.RKNNlib.converter.convert_darknet.Convolution_Blob_Assign.assign
E   File "/home/nationalflag/.local/lib/python3.6/site-packages/numpy/core/fromnumeric.py", line 292, in reshape
E     return _wrapfunc(a, 'reshape', newshape, order=order)
E   File "/home/nationalflag/.local/lib/python3.6/site-packages/numpy/core/fromnumeric.py", line 56, in _wrapfunc
E     return getattr(obj, method)(*args, **kwds)
E ValueError: cannot reshape array of size 3650443 into shape (1024,512,3,3)
done
--> Building model
Traceback (most recent call last):
  File "rknn_transform_tiny.py", line 27, in <module>
    rknn.build(do_quantization=True, dataset='./dataset_416x416.txt')
  File "/home/nationalflag/.local/lib/python3.6/site-packages/rknn/api/rknn.py", line 220, in build
    inputs = self.rknn_base.net.get_input_layers()
AttributeError: 'NoneType' object has no attribute 'get_input_layers'
回复

使用道具 举报

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

本版积分规则

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


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