Toybrick

标题: 加载darknet模型问题 [打印本页]

作者: 谁染枫林醉 时间: 2019-5-29 14:30
标题: 加载darknet模型问题
我加载了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模型，相关的代码怎么改？小白求指导。。

作者: troy 时间: 2019-5-30 09:00
输出的shape对不上，你这输出是172380，但是你reshape(1, 1917, 4)总共也才7668。你可以参考下yolov3帖子的后处理。

作者: 谁染枫林醉 时间: 2019-5-30 10:24

troy 发表于 2019-5-30 09:00
输出的shape对不上，你这输出是172380，但是你reshape(1, 1917, 4)总共也才7668。你可以参考下yolov3帖子的 ...

好的，谢谢

欢迎光临 Toybrick (https://t.rock-chips.com/)