基于caffe的图像分类(1)——制作train.txt和val.txt文件

最新推荐文章于 2024-06-25 16:00:43 发布

小咸鱼_

最新推荐文章于 2024-06-25 16:00:43 发布

阅读量1.6w

点赞数 3

分类专栏： caffe 文章标签：图像分类 caffe

本文链接：https://blog.csdn.net/sinat_30071459/article/details/51613304

版权

本文介绍了如何为基于Caffe的图像分类任务创建train.txt和val.txt文件。内容包括将图片按类别组织在文件夹中，重命名成统一格式，并提供了相关代码示例。此外，文章提及了后续步骤，即利用这些文件生成lmdb数据集和计算均值。

摘要由CSDN通过智能技术生成

做的标签文件如下：（无打乱）

image/Audi/000001.jpg 0
image/Audi/000002.jpg 0
image/Audi/000003.jpg 0
image/Audi/000004.jpg 0

image文件夹下是各类别文件夹，如下：

Audi等文件夹内放的是该类别的图片,最好先对这些图片重命名为统一格式，如000001.jpg，000002.jpg这种格式，重命名代码

参考http://blog.csdn.net/sinat_30071459/article/details/50723212

下面的代码只做了训练集和验证集，没有做测试集，可以自己修改代码做测试集。

clc;
clear;
%%下面生成顺序的trainval.txt和labels文件
%先设置train占数据集的百分比，余下部分为val
maindir='image\';
wf = fopen('trainval.txt','w');
lbf=fopen('labels.txt','w');
train_percent=0.9;%val_percent=1-train_percent

subdir = dir(ma