py-faster-rcnn+cpu配置并训练自己的数据
来源:互联网 发布:软件设计师考试书籍 编辑:程序博客网 时间:2024/05/16 14:51
py-faster-rcnn的配置具体参考github主页https://github.com/rbgirshick/py-faster-rcnn,并且下载时一定要使用--recursive递归下载。cpu模式下需要修改几个配置文件,具体可参考博客http://www.cnblogs.com/justinzhang/p/5386837.html
配置起来并不麻烦,但是运行./tools/demo.py --cpu --net zf 时出现以下两个错误
Error parsing text-formate caffe.NetParameter:47:11 Message type “caffe.LRNParameter" has no field named "engine" .
Error parsing text-formate caffe.NetParameter:232:14 Message type “caffe.PythonParameter" has no field named "param_str" .
百度了一下并没有找到这个问题的解决办法,问了一下网友,要么就是caffe的版本不对,要么就是python接口没有配置好。之前我将py-faster-rcnn和caffe-fast-rcnn分开来在windows上下载的,于是乎我重新clone --recursive了整个py-faster-rcnn,运行demo.py时还是出现同样的错误,那只能是python接口没配置成功,可是我使用import caffe时并没有报错,找了网上的各个caffe配置python的博客,我发现自己漏了一步。
make pycaffe 后添加环境变量
sudo gedit /etc/profile或者sudo vim /etc/profile ,注意gedit和vim 后面有空格
加入export PYTHONPATH=~/py-faster-rcnn/caffe-fast-rcnn/python:$PYTHONPATH
还有重要一步,要使文件生效
source /etc/profile
这时再运行demo.py --cpu --net zf 就成功了
---------------------------------------------------------训练自己的数据----------------------------------------------------------------------------
首先要准备数据集,我准备的是VOC2007格式的数据,网上有很多博客有介绍。我先用matlab版的faste-rcnn训练,在日志文件中出现smooth_l1_loss_layer.cpp:49 Not Implemented Yet 之后就再没动静了,搜了一下才发现这是报错了,faster-rcnn默认只能用GPU来训练,运行demo.py可以使用CPU。在运行python版本时直接提示只能用GPU训练,训练还是使用python版本的好。
在CPU下训练数据集,需要对py-faster-rcnn内的roi_pooling_layer和smooth_L1_loss_layer改为CPU版本,github上已经有人改了原代码使之能运行在CPU上。
并重新编译。可直接从github上下载替换。https://github.com/neuleaf/faster-rcnn-cpu
CPU训练时还有修改一些配置文件,可以参考http://blog.csdn.net/wjx2012yt/article/details/52197698
折腾了几天,终于可以在CPU上训练自己的数据了。
- py-faster-rcnn+cpu配置并训练自己的数据
- py-faster-rcnn训练自己的数据
- py-faster-rcnn + cpu安装及训练自己的数据集
- py-faster-rcnn + cpu安装及训练自己的数据集
- py-faster-rcnn+CPU训练自己的数据集(二)
- py-faster-rcnn+CPU训练自己的数据集(一)
- 使用py-faster-rcnn训练自己的数据
- py-faster-rcnn用自己的数据训练模型
- py-faster-rcnn用自己的数据训练模型
- 用py-faster-rcnn训练自己的VOC数据集
- 使用py-faster-rcnn训练自己的数据集
- py-faster-rcnn(caffe)配置和运行,训练自己的数据
- py-faster-rcnn算法caffe配置,训练及应用到自己的数据集
- 使用faster rcnn训练自己的数据(py-faster-rcnn )
- faster-rcnn安装配置,训练自己的数据,MATLAB,Ubuntu14
- 用自己的数据训练Faster-RCNN
- faster-rcnn安装,训练自己的数据
- Faster RCNN 训练自己的数据集
- Linux搭建下载器
- zigzag conversion 字符串的Z字形分离
- Cogs 728. [网络流24题] 最小路径覆盖问题
- 重新上路
- AndroidStudio中jniLibs目录自动生成
- py-faster-rcnn+cpu配置并训练自己的数据
- 【C语言】模拟实现strcpy()、strncpy()函数
- Note02
- python隐藏windows磁盘
- 第九章 CSS的继承、层叠、特殊性、重要性
- com.fasterxml.jackson.databind.JsonMappingException: No serializer found for
- 人脸检测+对齐之JDA
- Heartbleed心脏出血漏洞原理分析
- Android ART运行时无缝替换Dalvik虚拟机的过程分析