跑一些开源模型,会遇到这样的报错:
cpu-nms 报错
ModuleNotFoundError: No module named ‘utils.nms.cpu_nms’
解决方案:把 cpu_nms.pyx
(一般路径为 ./utils/nms/cpu_nms.pyx
) 替换为 cpu_nms.py
.
cpu_nms.py 下载
这样, cpu-nms 就可以用了
gpu-nms 报错
懒得处理,可以直接在 ./utils/nms_wrapper.py
注释掉 gpu 相关的,强制使用前面调好的 cpu
Ref