嘿嘿,最近博客更得有点勤啊,主要博主最近快毕业了,闲来无事就没事多学点东西把。。。
博主昨天把最难的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目录中查看生成的文件了。
注意的是,地址什么一定要相对应,不然会报各种奇怪的错误。