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