美文网首页
2018-04-21-FasterRCNN下.so文件的生成

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

作者: siyecao99999 | 来源:发表于2018-04-21 16:55 被阅读0次

介绍:.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

相关文章

网友评论

      本文标题:2018-04-21-FasterRCNN下.so文件的生成

      本文链接:https://www.haomeiwen.com/subject/lcbxlftx.html