版权声明:本文为博主原创文章,转载请一定附上博主原文链接,并署名转自ZeroZone零域。 https://blog.csdn.net/ksws0292756/article/details/83512208
https://github.com/amdegroot/ssd.pytorch/issues/154#issuecomment-384856547
将 detection.py 文件中第49行(行数可以因版本不同而不同):
if scores.dim() == 0:
continue
改为
if scores.size(0) == 0:
continue