Window+pycharm+faster rcnn代码实现

Window+pycharm+faster rcnn代码实现

记录一下自己的实现过程

代码下载及参考博客

windows下运行Faster R CNN算法（CPU）

我运行的环境是win10+python3.7

第一步
首先在github中下载源码，网址https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3 。

解压后在pycharm中打开（选择打开项目）

第二步
安装相关的依赖库，可以点开 requirements.txt 文件查看。（pycharm中可直接生成虚拟环境）

还要安装TensorFlow-cpu，貌似只支持1.几

第三步

在pycharm界面最下角点击终端。在这里运行！！！

进入 Faster-RCNN-TensorFlow-Python3-masterdatacocopythonAPI目录
在当前目录下依次运行 python setup.py build_ext --inplace
                                     python setup.py build_ext install
再进入lib/utils，   运行 python setup.py build_ext --inplace
#这步应该需要vs2015，百度下载即可

我的vs2015安装了个半成品，在运行时会报错，可参考下面两个链接

1.fatal error C1083: 无法打开包括文件: “io.h”: No such file or directory，下载了个window 10 SDK解决问题参考文章

2.link : fatal error LNK1158: 无法运行“rc.exe”，复制了两个文件到对应位置，不过我这两个文件的位置和文中不一样，我一阵瞎摸索

第四步
下载数据集，我参考的代码中的README，按照网址下载压缩包然后合并到同一个文件中，移动到指定目录，修改相应文件名
这里要注意的一点是 ：文件路径保持一致：dataVOCDevkit2007VOC2007

第五步
进入网站，同样下载解压
http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz
解压后，按**dataimagenet_weights** 的文件路径将解压后文件放入该文件夹下。
注意 ： 解压完的文件名为vgg_16.ckpt，需要重命名将其改为vgg16.ckpt

第六步

train.py文件，79行 tfconfig.gpu_options.allow_growth = True
                       改为 tfconfig.gpu_options.allow_growth = False

运行**“train.py"**文件即可。
再运行demo（我电脑跑不动，直接训练自己的数据集了）

训练自己的数据集，参考文章，文章很详细

大概就是找到自己的图片，重命名（我没有调整大小），然后下载labelimg，标注数据，修改相关文件

仅供大家参考，不要问我问题，因为我小白一枚，不会python，我最熟练的工具是百度