遇到了一个纠结了很久得问题就是:
Traceback (most recent call last):
File "./tools/demo.py", line 138, in <module>
_, _= im_detect(net, im)
File "/home/wolf/Desktop/FPN-master/tools/../lib/fast_rcnn/test.py", line 298, in im_detect
box_deltas = box_deltas * stds + means
ValueError: operands could not be broadcast together with shapes (1000,8) (84,)
起初以为是train.prototxt和test.prototxt文件中num_output数修改错误,重新修改训练多次还是出现同样得问题。
现在分析觉得应该是通道对不上,报错得原因看起来就是通道对不上。分析认为应该是运用的预训练模型中包含了21类,但是我的检测任务是单目标检测任务,只有目标和背景。
后续如何解决再记录。

在使用预训练模型进行单目标检测任务时遇到通道不匹配的问题,具体表现为ValueError异常。初步判断原因是预训练模型包含21类而当前任务仅涉及目标与背景两类,导致维度不一致。
1万+

被折叠的 条评论
为什么被折叠?



