ubuntu14.04 caffe 制作自己的数据

版权声明:坚持原创,禁止转载。 https://blog.csdn.net/SLAM_masterFei/article/details/80374752

嘿嘿,最近博客更得有点勤啊,主要博主最近快毕业了,闲来无事就没事多学点东西把。。。

博主昨天把最难的nvidia驱动这坑给跳过了,但是cuda下载老是失败。。。。。。fuck the nvidia!!!

今天博主想自己做caffe的数据集,来给出详细的步骤。

首先切换到我的caffe的主目录,然后

cd data
我在data目录下创建了一个叫myself的文件夹

我创建了如下的文件

我来说明一下,test文件放我们需要的图片,测试用。train文件夹存放我们训练用的文件。test和train的txt文件分别放对应文件夹的名字和图片的标签,我这里为了简单,只放了几张猫狗的图片,并且标签用0-1来表示。具体表示如下:

我们可以看到,这里的第一行为图片的id,然后加个空格,然后放对应图片的标签。最后保存。

需要注意的是,保存完请确认数据的行数,我之前不知怎么就是多一行,可能保存的时候光标换行了,所以一直提示找不到文件。。。

然后我们把caffe里面自带的creatimagenet的驱动文件打开

修改如下地方

修改好之后,我们进入caffe的主目录

然后

./data/myself/create_imagenet.sh

这样就可以在example/myself目录中查看生成的文件了。

注意的是,地址什么一定要相对应,不然会报各种奇怪的错误。

猜你喜欢

转载自blog.csdn.net/SLAM_masterFei/article/details/80374752