|
我加载了darknet模型,
ret = rknn.load_darknet(model = './yolov3-tiny.cfg', weight='./yolov3.weights')
虽然生成了rknn,但是出现了以下错误:
--> Running model
done
inference result: [array([[ 1.1962891 , 0.03408813, 0.61376953, ..., -8.15625 ,
-7.390625 , -6.6289062 ]], dtype=float32), array([[ 0.80322266, 0.07818604, -0.08508301, ..., -8.359375 ,
-7.8632812 , -7.125 ]], dtype=float32)]
Traceback (most recent call last):
File "dark-ssd.py", line 100, in <module>
predictions = outputs[1].reshape((1, NUM_RESULTS, 4))
ValueError: cannot reshape array of size 172380 into shape (1,1917,4)
请问关于加载darknet模型,相关的代码怎么改?小白求指导。。
|
|