您的位置:首页 > 编程语言 > Python开发

Faster-RCNN-tf-windows-python-pycharm 采用CPU复现代码

2020-06-04 06:01 896 查看

源码地址:https://github.com/rbgirshick/
py-faster-rcnn
论文地址Faster R-CNN
由于论文中的源码采用的是caffe框架,所以需要对源码进行修改,使其能够在tensorflow下运行

1. 本机环境:
windows
tensorflow 2.0
Python 3.7
OpenCV 3.4.2
easydict 1.9
cython 0.29.15
scipy 1.4.1
pycocotools 2.0
上述列举的几个包都是必须的要有的,其中几个本人亲自踩过的坑。
2. 程序采用的代码https://github.com/endernewton/tf-faster-rcnn.git
项目代码结构如下:

3. 下载VOC2007数据集
解压后内容如下: 然后直接将解压的文件VOCdevkit2007复制到…/data目录下。

4. 安装相关的包
安装tensorflow、Python等这些基础的工具就不介绍了。只介绍几个有坑的软件包。
cython:注意该包的名字不是cpython

pycocotools:首先需要下载windows版本的cocoapi,链接为https://github.com/philferriere/cocoapi,然后将下载的文件放在该项目下,通过通过cmd命令进入该文件下的PythonAPI文件中,执行python setup.py build_ext install
需要注意的是:必须下载的是windows版本的cocoapi,否则无法安装。

yaml:在…/lib/model/config.py文件中,需要导入yaml包,在pycharm中直接安装yaml包,该错误还会显示,通过安装pyyaml包,该错误才会消失

easydict:直接在pycharm或者conda中无法安装该包,需要去官网下载easydict的安装包,地址:https://pypi.org/project/easydict/#files,将该文件解压后,进入该目录,在conda环境下执行python setup.py install

5. 训练模型
根据https://blog.csdn.net/qq_38497266/article/details/86234055?utm_source=app博客的步骤一步一步执行
6. 按照上述博客的内容在执行trainval_net.py时会报错误
Process finished with exit code -1073741819 (0xC0000005)
解决步骤:
(1)下载vgg_16.ckpt(将其重名为vgg16.ckpy),然后放入…/data/imagenet_weights/目录下。
(2)此时参考该博客https://blog.csdn.net/qiao1025566574/article/details/81037908中的第六个错误修改。我的h5py的版本是2.8.0,pycharm和官网显示最新版本是2.10.0,尝试在pycharm更新版本失败,在conda下更新版本也失败,去官网下载2.9.0和2.10.0,然后在本地安装也失败,最终放弃更新,直接卸载,问题解决。。

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: