Faster-RCNN-tf-windows-python-pycharm 采用CPU复现代码
源码地址:https://github.com/rbgirshick/py-faster-rcnn论文地址:Faster R-CNN由于论文中的源码采用的是caffe框架,所以需要对源码进行修改,使其能够在tensorflow下运行1. 本机环境:windowstensorflow 2.0Python 3.7OpenCV 3.4.2easydict 1.9cython 0.29...
源码地址: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,然后在本地安装也失败,最终放弃更新,直接卸载,问题解决。。
开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!
更多推荐
所有评论(0)