按照github的readme执行,一般会先运行build.py文件编译nms和roi pooling或roi align。
但一般这一步都会出现一个问题:
ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead
网上其他的解决方案一般是:使用from torch.utils.cpp_extension import BuildExtension替换from torch.utils.ffi import create_extension,并且将code中的create_extension也全部替换掉,你们可以试试,但一般都会报错:
dist must be a Distribution instance
新坑又出现了……
其实原始的错误貌似是因为setuptools的版本太新了,所以最直接的解决办法就是降低版本。
还有一种解决办法可以参考https://github.com/facebookresearch/maskrcnn-benchmark/,extension的创建以及setup.py的作用均等同于老版的faster rcnn中的nms或roi pooling部分的build.py,当然具体代码我还没有研究,但有大佬已经提前做好了faster rcnn 1.0的代码,https://github.com/jwyang/faster-rcnn.pytorch/blob/pytorch-1.0/。