读取训练集,训练集为包含多张图片的文件夹,并设置三个标签,matlab实现
时间: 2024-03-25 11:42:11 浏览: 15
可以按照以下步骤实现:
1. 首先需要读取文件夹中的所有图片,可以使用`dir`函数获取文件夹中所有文件的信息,然后筛选出图片文件的路径。
```matlab
folder_name = 'train';
file_list = dir(folder_name);
img_list = {file_list(~[file_list.isdir]).name};
img_paths = fullfile(folder_name, img_list);
```
2. 接下来需要设置标签。可以根据文件名中的信息来设置标签。这里以文件名中是否包含`good`、`bad`和`normal`关键字来作为标签。
```matlab
labels = zeros(length(img_paths), 1);
for i = 1:length(img_paths)
if contains(img_list{i}, 'good')
labels(i) = 1;
elseif contains(img_list{i}, 'bad')
labels(i) = -1;
elseif contains(img_list{i}, 'normal')
labels(i) = 0;
end
end
```
完整代码如下:
```matlab
folder_name = 'train';
file_list = dir(folder_name);
img_list = {file_list(~[file_list.isdir]).name};
img_paths = fullfile(folder_name, img_list);
labels = zeros(length(img_paths), 1);
for i = 1:length(img_paths)
if contains(img_list{i}, 'good')
labels(i) = 1;
elseif contains(img_list{i}, 'bad')
labels(i) = -1;
elseif contains(img_list{i}, 'normal')
labels(i) = 0;
end
end
```