博客

先检查pytorch 和 torchvision 是否正确安装。

import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)

我是显示正常版本号，错误的话会显示false。显示false的情况这里不讨论。
也就是说正确安装了，那就去ssd\utils\nms.py里，修改以下代码：

import torch
import torchvision

if torchvision.__version__ >= '0.0.0': # 之前是0.3.0
    _nms = torchvision.ops.nms
else:
    warnings.warn('No NMS is available. Please upgrade torchvision to 0.3.0+')
    sys.exit(-1)

原因是python比较版本号的时候，按照字符下标顺序比较，如：0.13.1和0.3.0，比较的时候是13的1和3进行比较，就会报错。