2018-04-21-FasterRCNN下.so文件的生成

介绍:.so文件类似于dll,可直接调用

1. 基于xx_kernel.cu 和xx.c 生成_rroi_pooling.so动态链接文件的过程

Rotated Roi_Pooling(cuda实现)代码来源:https://github.com/siyecao99999/RRPN/tree/master/caffe-fast-rcnn/src/caffe/layers/rotate_roi_pooling_layer.cu

(1)仿照FasterRCNN中roi_pooling_kernel.cu编写rroi_pooling_kernel.cu文件,主要是添加函数RotateROIPoolForward和RotateROIPoolBackward

(2)使用编译命令nvcc -c -o rroi_pooling.cu.o rroi_pooling_kernel.cu -D GOOGLE_CUDA=1 -x cu -Xcompiler -fPIC -arch=sm_52生成rroi_pooling.cu.o文件

(3)运行roi_pooling下的build.py文件,在_ext/rroi_pooling下生成文件_rroi_pooling.so

2. 基于xx_kernel.cu和xx.pyx生成rbbox_overlaps.so和rotate_polygon_nms.so动态链接文件的过程

rotated相关代码来自https://github.com/siyecao99999/RRPN/tree/master/lib/rotation

(1) 在FasterRCNN中的setup.py里面添加相关的Extension实例,如下所示:

(2)运行命令python setup.py build_ext --inplace

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容