出现IndexError: The shape of the mask [32, 8732] at index 0 does not match the shape of the indexed tens解决办法:
在multibox_loss.py的第97行
loss_c[pos] = 0 # filter out pos boxes for now
的前面加一行
loss_c = loss_c.view(pos.size()[0], pos.size()[1])
主要是记录一下自己平时遇到的问题,和大家分享一下
如有侵犯,请联系我
点个赞支持一下吧