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

男娘i 2022-05-15 14:18 500阅读 0赞

https://github.com/amdegroot/ssd.pytorch/issues/154#issuecomment-384856547

将 detection.py 文件中第49行(行数可以因版本不同而不同):

  1. if scores.dim() == 0:
  2. continue

改为

  1. if scores.size(0) == 0:
  2. continue

发表评论

表情:
评论列表 (有 0 条评论,500人围观)

还没有评论,来说两句吧...

相关阅读