windows 下编译py-faster-rcnn，py-rfcn， py-pva-frcn 下的lib。编译Cython模块

最新推荐文章于 2024-08-06 15:05:04 发布

置顶

rookie_chenzhi

最新推荐文章于 2024-08-06 15:05:04 发布

阅读量1.8w

点赞数 13

本文链接：https://blog.csdn.net/chenzhi1992/article/details/53374265

版权

在Windows上运行py-faster-rcnn、py-rfcn和py-pva-frcn需要编译lib目录下的setup.py来创建Cython模块。由于Windows与Linux环境的不同，直接在cmd中编译会失败。解决方案是修改setup.py，将其分为newsetup.py和setup_cuda.py。newsetup.py用于编译非GPU部分，而setup_cuda.py处理包含.cu文件的GPU部分。编译过程包括在cmd中分别运行这两个文件的指令。若缺少gpu_nms.cu文件，可在指定QQ群获取。

摘要由CSDN通过智能技术生成

windows下想要运行py-faster-rcnn，py-rfcn， py-pva-frcn ，首先就是要编译lib文件夹下的setup.py ，建立Cython模块。其实这一些方法的lib都差不多，只需对照原版的setup.py进行相应的微改，记得修改代码中相应的文件夹名字。在Linux下编译这个很简单，在windows下面直接在cmd下编译是不会通过的，需要修改setup.py。这里提供修改后的版本（windows下），将setup.py 分成两个文件：newsetup.py 和 setup_cuda.py（因为lib里面有.cu 文件需要编译，针对使用gpu的，如果是用cpu的话，就不需要编译这个文件了。）

（有问题可以加群交流334227548）

这里提供pva-faster-rcnn版本的两个文件（其他算法的lib下的setup.py 相似。）

1 先编译newsetup.py

cmd 输入如下命令：

如果需要如下问题：

cmd中输入如下指令：

SET VS90COMNTOOLS=%VS110COMNTOOLS% （如果用的是vs2012的话）
SET VS90COMNTOOLS=%VS120COMNTOOLS% （如果用的是vs2013的话）

编译成功截图：

下面提供newsetup.py 的代码：

# --------------------------------------------------------
# Fast R-CNN
# Copyright (c) 2015 Microsoft
# Licensed under The MIT License [see LICENSE for details]
# Written by Ross Girshick
# ------------------------