matlab imagedatastore支持五种图像标注方法:一、按子文件夹名自动标注;二、用正则表达式从文件名提取标签;三、通过外部csv/mat文件映射标签;四、混合文件夹与文件名生成复合标签;五、过滤无效图像并同步更新标签。

如果您需要为深度学习模型准备图像数据,但手动逐张读取、组织文件夹并分配标签耗时易错,则可能是由于缺乏高效的数据管理机制。以下是使用Matlab imageDatastore 实现批量读取与自动标注的具体操作步骤:
一、基于文件夹名称自动标注
该方法利用子文件夹的命名直接生成类别标签,适用于标准分类数据集结构(如每个类别一个独立子文件夹)。imageDatastore在创建时扫描所有子文件夹路径,并将文件夹名转换为categorical类型的标签,无需额外代码干预。
1、确保数据集目录结构符合规范:根目录下包含多个子文件夹,每个子文件夹名称即为类别名,例如:Cat、Dog、Bird;各子文件夹内存放对应类别的图像文件。
2、在Matlab中执行以下命令创建带标签的数据存储对象:imds = imageDatastore('D:\PetImages','IncludeSubfolders',true,'LabelSource','foldernames');
3、验证标签是否正确生成:在命令行输入imds.Labels,观察输出是否为categorical数组,且值与子文件夹名称一致。
二、按文件名规则提取标签
当图像文件名中嵌入类别信息(如“cat_001.jpg”、“dog_002.png”),而文件夹结构不统一时,可借助正则表达式从文件名中解析标签。此方式绕过文件夹层级依赖,增强对非标准数据组织的兼容性。
1、定义正则表达式模式以捕获类别关键词,例如:pattern = '^(cat|dog|bird)_'。
2、获取全部文件路径:allFiles = imds.Files;
3、遍历路径列表,用regexpi提取匹配项,并构造标签数组:labels = cell(numel(allFiles),1); for i = 1:numel(allFiles), match = regexp(allFiles{i}, pattern, 'tokens'); labels{i} = match{1}{1}; end
4、将自定义标签赋值给数据存储:imds.Labels = categorical(labels);
三、使用外部标签文件映射标注
当存在独立CSV或MAT文件记录每张图像的类别信息时,可通过外部元数据驱动标注过程。该方法支持细粒度控制,适用于含多标签、属性字段或版本化标注的数据集。
1、准备标签表文件labelMap.csv,包含两列:FileName(不含路径的纯文件名)和Class(类别名称)。
2、读取CSV生成表格:labelTable = readtable('labelMap.csv');
3、提取imageDatastore中各文件的基础名:baseNames = cellfun(@(x) fileparts(x){2}, imds.Files, 'UniformOutput', false);
4、通过ismember匹配文件名并索引对应标签:[~, idx] = ismember(baseNames, labelTable.FileName); imds.Labels = categorical(labelTable.Class(idx));
四、混合标注策略:子文件夹+文件名增强
在复杂场景中,单一标注源可能不足以覆盖全部语义需求。例如主类别由文件夹决定,子类别或属性由文件名后缀体现。此时可组合两种信息构建复合标签,提升数据表达能力。
1、先启用基础文件夹标注:imds = imageDatastore('D:\Dataset','IncludeSubfolders',true,'LabelSource','foldernames');
2、提取当前标签与对应文件路径:currentLabels = imds.Labels; filePaths = imds.Files;
3、针对每个路径,解析文件名中的附加标识符(如“_blurry”、“_night”):suffixes = regexpi(filePaths, '_([a-z]+)\.jpg$', 'tokens');
4、拼接主标签与后缀形成新标签:enhancedLabels = arrayfun(@(l,s) strcat(l,'_',s{1}), currentLabels, suffixes, 'UniformOutput', false); imds.Labels = categorical(enhancedLabels);
五、过滤无效图像并同步更新标签
数据集中常混杂损坏、非图像格式或尺寸异常的文件,若未剔除将导致后续读取失败。imageDatastore支持预过滤机制,在初始化阶段即排除问题文件,并自动保持路径与标签的一致性。
1、指定允许的图像扩展名:validExts = {'.jpg','.jpeg','.png','.bmp'};
2、创建数据存储时加入过滤参数:imds = imageDatastore('D:\Dataset','IncludeSubfolders',true,'LabelSource','foldernames','FileExtensions',validExts);
3、检测并移除无法解码的图像:ok = zeros(size(imds.Files)); for i = 1:length(imds.Files), try, readimage(imds,i); ok(i) = 1; catch, end; end imds = subset(imds, logical(ok));











