SSD pytorch 源码demo报错: ValueError: not enough values to unpack (expected 2, got 0)

版权声明:本文为博主原创文章,转载请一定附上博主原文链接,并署名转自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

猜你喜欢

转载自blog.csdn.net/ksws0292756/article/details/83512208