windows10编译RRPN/R2CNN(rotate)
windows10编译Faster RCNN/RRPN/R2CNN所需的setup.py(rotate)
环境:
miniconda with python3.7.6
vs2015
win10
pytorch 1.2
使用pytorch里的 BuildExtension, CUDAExtension,否则不成功
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
需要修改的地方:
rotate_gpu_nms.cpp 和r
原创
2020-05-16 00:12:24 ·
1329 阅读 ·
3 评论