caffe对自定义数据集进行分类

最新推荐文章于 2025-03-08 13:45:37 发布

小白aaa

最新推荐文章于 2025-03-08 13:45:37 发布

阅读量2.1k

点赞数

分类专栏： caffe学习文章标签： caffe

本文链接：https://blog.youkuaiyun.com/u013126125/article/details/52851898

版权

1.准备数据集:

/examples/imagenet/新建myimage文件夹存放数据集

myimage，train(bus,dinosaur,elephant,flower,horse)五个类别的训练集,每类80张，val(......)五个类别的测试集，每类20张制作标签:其中matlab代码:

  pos_folder='/home/handson/caffe/examples/imagenet/myimage/val/';
  pos = dir(pos_folder);
  fid=fopen('val.txt','w'); 
  for i=3:7
           img_name = [pos(i).name];
           img_folder=[pos_folder,'/',pos(i).name];
            images=dir(img_folder);
        
       for j=3:size(images,1)
           d=strcat(img_name,'/',images(j).name);
           fprintf(fid,'%s %d\n',d,i-3);//这里%s和%d之间只有一个空格，不然下面生成数据库有误
       end
     end
     fclose(fid);