pytorch ssd github : https://github.com/amdegroot/ssd.pytorch
运行demo中的demo.ipynb时,出现 need more than 0 values to unpack的错误。
解决方法:
将ssd.pytorch/layers/functions/detection.py第 49 行修改为:
# if scores.dim() == 0:
if scores.size(0) == 0:
continue
然后需要重启jupyter notebook 重新运行。即可成功。