代码地址:https://github.com/jwyang/faster-rcnn.pytorch.git
编译前提
需要安装显卡驱动、cuda和cudnn。
我使用的是pycharm建的一个环境python3.6的环境。
注意:你需要要安装requirements.txt里面所有的依赖,这里的依赖里面没有提到pytorch,你可以代码主页发现他的master-branch的pytorch版本是0.4.0,作者也提供了pytorch 1.0.0版本的代码,不过需要切换一下分支,即https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0。
命令行操作:
cd faster-rcnn.pytorch
cd build
sh make.sh
这时你发现系统没用调用你创建的python环境,那是因为没有设置系统默认python环境,有很多中设置方法,这里写一下我自己使用的一种:
gedit ~/.bashrc
在.bashrc最后添加自己python的位置目录:
export PATH=/home/xinyu/pycharm/venv/bin/${PATH:+:${PATH}}
保存后在
source ~/.bashrc
</