Toybrick

加载darknet模型问题

谁染枫林醉

注册会员

积分
74
发表于 2019-5-29 14:30:33    查看: 6545|回复: 2 | [复制链接]    打印 | 显示全部楼层
我加载了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

版主

积分
2346
发表于 2019-5-30 09:00:38 | 显示全部楼层
输出的shape对不上,你这输出是172380,但是你reshape(1, 1917, 4)总共也才7668。你可以参考下yolov3帖子的后处理。
回复

使用道具 举报

谁染枫林醉

注册会员

积分
74
 楼主| 发表于 2019-5-30 10:24:07 | 显示全部楼层
troy 发表于 2019-5-30 09:00
输出的shape对不上,你这输出是172380,但是你reshape(1, 1917, 4)总共也才7668。你可以参考下yolov3帖子的 ...

好的,谢谢
回复

使用道具 举报

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

本版积分规则

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


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