编译nms

编译nms:

ModuleNotFoundError: No module named' xxx.nms.cpu_nms'
说明缺少编译的nms
在 https://github.com/hanlinshi/nms 下载nms代码
这个里面包含了两个版本的nms里面的函数有所不同。自己可以查看一下。选择一个下载进行编译
进入目录进行编译

$ python3 setup.py build_ext --inplace

运行以后会得到两个so文件,一个cpu_xxxx.so 一个 gpu_xxxx.so
将这两个文件拷贝到项目中的nms目录下即可。

你可能感兴趣的:(问题,pytorch,问题)