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

请问楼主的rknn-toolkit是哪个版本？ 我在1.0.0和1.2.1版本上，python3 rknn_transform_320x320.py 出现错误：
--> Loading model
E Catch exception when loading darknet model: ./yolov3_320x320.cfg!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 380, in rknn.api.rknn_base.RKNNBase.load_darknet
E   File "rknn/base/RKNNlib/converter/convert_darknet.py", line 471, in rknn.base.RKNNlib.converter.convert_darknet.Converter.load_blobs
E   File "rknn/base/RKNNlib/converter/convert_darknet.py", line 215, in rknn.base.RKNNlib.converter.convert_darknet.Convolution_Blob_Assign.assign
E   File "<__array_function__ internals>", line 6, in reshape
E   File "/home/abc/.local/lib/python3.6/site-packages/numpy/core/fromnumeric.py", line 301, in reshape
E     return _wrapfunc(a, 'reshape', newshape, order=order)
E   File "/home/abc/.local/lib/python3.6/site-packages/numpy/core/fromnumeric.py", line 61, in _wrapfunc
E     return bound(*args, **kwds)
E ValueError: cannot reshape array of size 148814 into shape (256,128,3,3)
done

appleUncle 发表于 2019-12-25 10:22
请问楼主的rknn-toolkit是哪个版本？ 我在1.0.0和1.2.1版本上，python3 rknn_transform_320x320.py 出现错 ...

自己解决了，是yolo的wieghts下载不全，重新下载就好了
cd yolov3_demo
wget https://pjreddie.com/media/files/yolov3-tiny.weights
wget https://pjreddie.com/media/files/yolov3.weights