CSP运行出现No module named gpu_nms解决办法

CSP源码:https://github.com/liuwei16/CSP

CSP论文:https://arxiv.org/abs/1904.02948

CSP算法思想:https://www.jiqizhixin.com/articles/2019-04-13

在运行test code时出现No module named gpu_nms错误。

原因:作者公布的源码并没有make编译gpu_nms模块

解决方法:参照Faster R-CNN中gpu_nms编译方法,具体步骤如下:

1. 下载Faster R-CNN源码,地址:https://github.com/rbgirshick/fast-rcnn

2. 将下载的代码中py-faster-rcnn/lib/目录下除nms文件夹之外的所有文件(夹)复制到CSP/keras_csp/中;

3. 然后cd到keras_csp下执行指令:make


版权声明:本文为xunan003原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。