win7下配置faster-rcnn的tensorflow版本(gpu版本)

首先给出源代码网址:https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5


所需的软件环境:python3.5+tensorflow-gpu1.40(gpu版本)+vs2015+pycharm+cuda8.0+cudnn6.1


cuda8.0下载地址:https://developer.nvidia.com/cuda-80-ga2-download-archive

cudnn6.1下载地址:https://developer.nvidia.com/cudnn

cuda8.0和cudnn很好安装,安装自行百度教程安装即可。


python3.5下载地址:https://www.python.org/downloads/release/python-352/

python的其他库安装(用pip3即可):

cython, python-opencv, easydict......



tensorflow安装方法:

tensorflow-gpu1.40下载地址:https://pypi.python.org/pypi/tensorflow-gpu/1.4.0

tensorflow-cpu1.40下载地址:https://pypi.python.org/pypi/tensorflow/1.4.0

安装方法和python安装其他库的方法相同,即在命令行下使用pip3


faster-rcnn安装方法:

1.  Go to ./data/coco/PythonAPI

      Run python setup.py build_ext --inplace

      Run python setup.py build_ext install

      环境配置完毕

2. 下载数据集

       下载地址:

http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar

        下载之后放在:

        "data/VOCDevkit2007/annotations_cache"

        "data/VOCDevkit2007/VOC2007"

3. 下载预训练好的vgg16网络或者其他网络

        下载地址:https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models


        将下载好的模型放在:$faster-rcnn目录$\data\imagenet_weights下面

4. 训练模型

        运行 train.py程序即可,一般用cpu跑迭代一次需要20s,用gpu跑一次迭代需要2-3s

        每经过5000次迭代,会保存网络相关文件,保存的信息在:$faster-rcnn目录$\default\voc_2007_trainval\default


5. 运行demo.py

        将步骤4中保存的文件放在   下面,运行demo.py便会出现结果




  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 18
    评论
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值