Faster-rcnn-matlab-cuda8.0+ZFnet训练自己的数据

主要参考这篇博文的,一步步配置就能跑起来~~

先说一下所需要的硬件吧:我一开始用的是GT630,只有2G显存,程序跑到一半报错out of memory才知道起码要3G显存才能训练ZF net,至于VGG-16 net更是需要高达8G的显存。于是,换了台实验室的机子,用GTX1080试试。

至于软件,win7系统 + matlab 2014b + cuda 8.0 + vs 2013,别的没什么好说的,cuda 8.0 是1080的标配,不用不行。用了cuda8.0的缘故,mex文件需要自己编译,这也是个巨坑,不过最好还是学会自己编译,省的以后版本变了自己还要去网上满世界的找别人编译好的成品。

那就先讲讲vs2013+cuda8.0编译caffe的external文件吧~(vs与cuda的安装很无脑,不讲了,建议先装vs再装cuda)

编译过程参考这里:Caffe for Faster R-CNN,按步骤做就行了。

看上去很简单是吧?操作起来一大堆坑!!

1)下载caffe_library

2)将github上的caffe拷贝到caffe_library的caffe目录下

3)准备opencv,boost,mkl

a)boost不管什么版本一定要注意下载x64位的安装包,安装完之后会有 lib64-msvc-12.0 这么一个文件夹后面会用到。

网上说一定要编译才能用,不过实际上lib64-msvc-12.0 已经预编译了我们需要的所有库,直接用即可。

b)opencv使用未编译包;

c)mkl需要安装

4)打开caffe_library目录中的slu,更改配置为Release_Mex

打开的时候发现项目caffe无法加载,那是因为默认的是cuda6.5,所以要右键caffe项目,修改配置文件中cuda版本号

5)修改include路径和lib路径

要怎么操作可以百度,路径到底具体到哪个子文件夹可以参考作者的(作者已经添加了自己电脑上的外部库路径,你只需要稍微做一下修改即可)

6)设置caffe为启动项目

7)重新编译caffe包

这儿也有个坑,由于boost已经用了C++14的标准,而VS2013要到update5才支持C++14,所以没办法只能乖乖地升级到update5。

8)复制 .\x64\Release_Mex 中的所有文件到 faster_rcnn-master\external \caffe\matlab\caffe_faster_rcnn.

这里有我编译好的库,如果跟我的环境一样的话可以拷贝过去

到此为止,faster-rcnn的external库更新完毕~~~~~~

撒花~~~

然后运行faster_rcnn-master\faster_rcnn_build.m 不出意料,报错!

这时候找到.\function\nms\nvmex.m文件,修改Host_Copliler_Location为你电脑上vs2012的路径,重要是事说三遍!

!2012!2012!2012!

也就是说你还要安装vs2012,反正我的2013不行,不知道还有么有别的办法。

编译完mex文件意味着预备工作完成了。

接下来要准备自己的数据集,可以参考这篇博客,里面要用的工具可以从这里下载

亲测标注工具只能在win7 下使用,win8 、win10都不行。

按照教程一一修改文件,运行experiments/script_faster_rcnn_VOC2007_ZF.m 很快就能进入迭代了。

现在我的1080正在高速运转,估计明天上午就能出结果了……

时间: 2024-11-10 18:52:27

Faster-rcnn-matlab-cuda8.0+ZFnet训练自己的数据的相关文章

faster r-cnn 在CPU配置下训练自己的数据

因为没有GPU,所以在CPU下训练自己的数据,中间遇到了各种各样的坑,还好没有放弃,特以此文记录此过程. 1.在CPU下配置faster r-cnn,参考博客:http://blog.csdn.net/wjx2012yt/article/details/52197698#quote 2.在CPU下训练数据集,需要对py-faster-rcnn内的roi_pooling_layer和smooth_L1_loss_layer改为CPU版本, 并重新编译.这位博主对其进行了修改,可直接进行替换:htt

Faster RCNN代码理解(Python)

转自http://www.infocool.net/kb/Python/201611/209696.html#原文地址 第一步,准备 从train_faster_rcnn_alt_opt.py入: 初始化参数:args = parse_args() 采用的是Python的argparse 主要有–net_name,–gpu,–cfg等(在cfg中只是修改了几个参数,其他大部分参数在congig.py中,涉及到训练整个网络). cfg_from_file(args.cfg_file) 这里便是代用

Faster RCNN算法代码解析

一. Faster-RCNN代码解释 先看看代码结构: Data: This directory holds (after you download them): Caffe models pre-trained on ImageNet Faster R-CNN models Symlinks to datasets demo 5张图片 scripts 下载模型的脚本 Experiments: logs scripts/faster_rcnn_alt_opt.sh cfgs/faster_rcn

FCOS及其和Faster R-CNN的区别

RetinaNet,SSD,YOLOv3,Faster R-CNN等都是Anchor-based的检测器,即需要预定义的Anchor boxes来进行训练.FCOS是一种Anchor-free和Proposal-free的检测器,即不需要预定义Anchor boxes来进行训练,从而节省了对计算资源的占用. 根据我的理解,FCOS和Faster R-CNN的区别主要表现在: (1) Faster R-CNN是每个点提出N个Proposal,N一般等于9(预设了3种长宽比例,3种尺寸因此是9个Pr

ubuntu16.04 安装配置matlab ,python ,cuda8.0,cudnn,opencv3.1的caffe环境

网络上有很多ubuntu上caffe配置环境的帖子,本人照着其中的许多进行了参考,都出现了或多或少的错误,很多地方也有差异. 于是自己整理了下自己的安装过程,成功进行了测试,跑通了faster-rcnn.配置环境时间为2017.1.4 系统ubuntu16.04 一:显卡驱动的安装: 由于要使用GPU,所以先要查看自己显卡所匹配的显卡驱动,网址:http://www.nvidia.com/Download/index.aspx%3Flang=en-us 选择电脑匹配的显卡驱动,本人电脑显卡为GT

Faster R-CNN利用新的网络结构来训练

前言 最近利用Faster R-CNN训练数据,使用ZF模型,效果无法有效提高.就想尝试对ZF的网络结构进行改造,记录下具体操作. 一.更改网络,训练初始化模型 这里为了方便,我们假设更换的网络名为LeNet. 首先,需要先训练在Faster R-CNN中用来初始化网络的模型:LeNet.caffemodel. 这里比较简单,直接用完整的LeNet去训练一部分数据(VOC2007,VOC2012均可),数据初始大小resize为224*224,即可得到初始化网络的模型. 二.在Faster R-

caffe 在window下编译(windows7, cuda8.0,matlab接口编译)

1. 环境:Windows7,Cuda8.0,显卡GTX1080,Matlab2016a,VS2013   (ps:老板说服务器要装windows系统,没办法,又要折腾一番) 2.下载caffe 包:https://github.com/BVLC/caffe/tree/windows 按照Github上的命令行就行: 这里的Projects是自己在本地新建的文件夹,随便放在哪里,如果git没有安装的话,先安装git,百度git即可. 执行前3个命令之后,就把caffe包下载下来了. 在运行bui

Widows下Faster R-CNN的MATALB配置(GPU)

目录 1. 准备工作 2. VS2013编译Caffe 3. Faster R-CNN的MATLAB源码测试 说在前面,这篇是关于Windows下Faster R-CNN的MATLAB配置,GPU版本:CPU版本见:Widows下Faster R-CNN的MATALB配置(CPU): 相比较来说,CPU版本相对容易一些,因为涉及到GPU的计算,还需要配置CUDA以及CuDNN: 下面将分成三个部分进行介绍: 准备工作:下载caffe.faster rcnn源码.安装cuda.cudnn: VS2

【论文笔记】Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks

写在前面: 我看的paper大多为Computer Vision.Deep Learning相关的paper,现在基本也处于入门阶段,一些理解可能不太正确.说到底,小女子才疏学浅,如果有错误及理解不透彻的地方,欢迎各位大神批评指正! E-mail:[email protected]. ------------------------------------------------ <Faster R-CNN: Towards Real-Time Object Detection with Reg