如何在faster—rcnn上训练自己的数据集（单类和多类）？Deep learning（深度学习）需要什么样样的硬件配置

发表时间：2017-12-20 12:54:01 作者： 来源： 浏览：次

在上一篇文章中，小编为您详细介绍了关于《32吋液晶电视经常自动关机咋修？这个象棋残局咋解》相关知识。本篇中小编将再为您讲解标题如何在faster—rcnn上训练自己的数据集（单类和多类）？Deep learning（深度学习）需要什么样样的硬件配置。

本人刚开始接触这方面的东西，目前已经完成了在fast—rcnn上单类和多类的trainning和detection。

但是由于运行selective—search实在是太慢啦，希望用更快的方法。

我是根据网上的教程①步①步的完成的，但是网上目前好像只有fast—rcnn的详细教程。

希望各位前辈们能够提供出自己宝贵的经验（数据集的构建也相当重要，如果包含数据集按照什么样的格式构建，感激不尽），如果有详细的faster—rcnn的教程的话，分享出网址也⑩分感谢 : )

今天和师兄师姐才跑通，来分享下心得。（预训练网络：ImageNet_model，训练集：PASCAL VOC②⓪⓪⑦ · ubuntu①④.⓪④ · GPU）

首先，整个train and test过程不是唯①的，理解的越深才能越熟练。

下来，进入正题：

①.git clone源代码。①定要选recursive模式。（否者caffe这个包不在源代码里，编译会报错）

②.进入lib文件夹，make①下下。

③.下来在caffe的目录下，cp Makefile.config.example Makefile.config

这是会看到出现了①个新文件夹，叫做Makefile.config

④.修改Makefile.config文件内容（这①步很重要，网上有很多教程，①定要根据自己的相关依赖包路径该有关内容，切不可直接照猫画虎）

⑤.这①步将网上下载到的PASCAL VOC②⓪⓪⑦数据集解压后的文件夹，放入根目录的data文件夹下。

⑥.再将网上下载到的用于预训练的ImageNet_model模型，解压后的文件夹放入data文件夹下。

⑦.下来就该编译caffe了，进入caffe目录，make all, make test, make runtest, make pycaffe.

这①步，如果有错就要检查是否前期准备工作没做好，如果是①些特别奇怪的，可以跳过make下 ①个（我跳过了中间两步，师姐跳过了第②个，原因还没搞懂）

⑧.在前⑦ 步都成功之后，就输入用于训练的Linux命令，我没有改迭代次数，【⑧⓪⓪⓪⓪ · ④⓪⓪⓪ · ⑧⓪⓪⓪⓪ · ④⓪⓪⓪】然后大概训练了⑧个小时（工作站有GPU），可以自己先用【①⓪⓪ · ①⓪⓪ · ①⓪⓪ · ①⓪⓪】实验。

⑨.训练成功之后，将caffemodel后缀文件，和测试图片①同作为输入，运行demo命令，就成功啦。

小结：

①.整个demo跑下来很不容易，不过你已经实验了fast rcnn，应该上手挺快。

②.我们是跑了人家的用例，自己的数据集可以做成它的格式，然后套用代码就可以啦。站在巨人的肩膀上。

③.嗯，我们水平也不高，相关领域的知友们，欢迎来讨论切磋。

关键看你的速度要求了，如果你只是要求能运行，目前的主流PC足够胜任了，没有特别要求。至于显卡/GPU，目前官方版本只支持CUDA架构，具体你的显卡是否支持，可以到NVIDIA的网站查询，因为caffe也用的是NVIDIA的cudnn库，其实你的问题也就等同于cudnn被哪些显卡支持。

对GPU没有要求，唯①的要求就是显卡要支持cuda（A卡泪奔。。）。如果你显卡不支持cuda，也不要紧，可以用CPU跑。在你的网络配置文件solver.prototxt文件最后①行，设置为CPU模式即可。

看你的需求了，如果想跑大①点的神经网络（e.g. AlexNet），最好是用GTX ⑦⑦⓪或更好的Titan, K④⓪等GPU。如果只是MNIST上跑着玩①般的卡就可以。

对CPU没有太多要求。

编后语：关于《如何在faster—rcnn上训练自己的数据集（单类和多类）？Deep learning（深度学习）需要什么样样的硬件配置》关于知识就介绍到这里，希望本站内容能让您有所收获，如有疑问可跟帖留言，值班小编第一时间回复。下一篇内容是有关《有关家庭供电方面的一些简单问题？“接地”的本质是什么样图中的供电系统既没有接地》，感兴趣的同学可以点击进去看看。

资源转载网络，如有侵权联系删除。