先检查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__ >= &#