在复现论文《Mask R-CNN》中,程序用到了PyTorch中的一个包(torch.utils.ffi),但是博主安装的PyTorch版本是1.12.1,也就因此导致了上面的错误。
原因:
在较新版的PyTorch中,torch.utils.ffi被弃用了,需要用其他包来替代。
解决办法:
对于遇到的问题,将原语句:
from torch.utils.ffi import create_extension
修改成:
from torch.utils.cpp_extension import BuildExtension
再将文件下面的调用:
ffi = create_extension( '_ext.nms',
headers=headers,
sources=sources,
define_macros=defines,
relative_to=__file__,
with_cuda=with_cuda,
extra_objects=extra_objects
)
改成:
ffi = BuildExtension(
'_ext.nms',
headers=headers,
sources=sources,
define_macros=defines,
relative_to=__file__,
with_cuda=with_cuda,
extra_objects=extra_objects
)
就可以了!