SSD pytorch 源码demo报错: ValueError: not enough values to unpack (expected 2, got 0)
https://github.com/amdegroot/ssd.pytorch/issues/154#issuecomment-384856547
将 detection.py 文件中第49行(行数可以因版本不同而不同):
if scores.dim() == 0:
continue
改为
if scores.size(0) == 0:
continue
还没有评论,来说两句吧...