CIFAR10数据集Matlab读取、保存、可视化代码

最近做项目要用到cifar10数据集，想看下里面的图片，因为平时比较熟悉matlab，就想在matlab里直接显示出数据集中的图片，但是在网上找了很多相关的代码，无一列外，都无法做到实时显示，非常的不方便，后面我就准备自己整了！下面这段代码保证亲测有效，直接复制到matlab中，就可以看到啦！

一、前情提要

CIFAR-10 数据集包含10个不同的类别，每个类别都用一个数字表示。这些数字标签对应于数据集中的不同物体或场景类别，例如飞机、汽车、鸟类、猫、狗等。因此CIFAR-10 数据集中的标签通常是数字。

以下是CIFAR-10 数据集中的10个类别及其对应的数字标签：

1. 飞机 (airplane) - 标签 0
2. 汽车 (automobile) - 标签 1
3. 鸟类 (bird) - 标签 2
4. 猫 (cat) - 标签 3
5. 鹿 (deer) - 标签 4
6. 狗 (dog) - 标签 5
7. 青蛙 (frog) - 标签 6
8. 马 (horse) - 标签 7
9. 船 (ship) - 标签 8
10. 卡车 (truck) - 标签 9

在机器学习任务中，这些数字标签通常用于表示每个图像所属的类别。例如，如果标签为3，那么对应的图像被视为属于猫这个类别。这些标签是训练机器学习模型以识别图像类别所使用的目标值。

二、CIFAR10数据集的下载

我直接就是一个压缩包：（密码：shfah8）

CIFAR10数据集​share.weiyun.com/iiG1Rj8i

数据集中一共50000张彩色图片，它给分为了五个批次，一个批次中有10000张，图片大小是32*32的。

                                                                 数据集中各文件

三、CIFAR10在Matlab中的读取、显示

①代码：这段代码实现的功能是读取CIFAR10数据集中前10000中的前10张，在matlab中显示出来图像，并且保存到数据集所在文件夹下的save_images文件夹中。

clc
clear all
% 指定 CIFAR-10 数据集的文件夹路径
dataFolder = '/path/to/cifar-10-batches-mat/';    %%

% 读取 data_batch_1 中的图像数据
data = load(fullfile(dataFolder, 'data_batch_1.mat'));
images = permute(reshape(data.data', [32, 32, 3, 10000]), [2, 1, 3, 4]);
% images = images(:, :, :, 1:10); % 仅选择前10张图像
labels = data.labels(1:10); % 获取对应的标签

% 创建文件夹以保存图像
outputFolder = dataFolder;
outputFolder = fullfile(outputFolder, 'saved_images');
if ~exist(outputFolder, 'dir')
    mkdir(outputFolder);
end

% 保存前10张图像为 .png 文件
for i = 1:10
    % 创建文件名
     fileName = fullfile(outputFolder, ['cifar10_image_label_' num2str(labels(i)) '_' num2str(i) '.png']);
    
    % 获取要保存的图像
    imageToSave = images(:,:,:,i);
    
    % 归一化像素值到 [0, 1] 范围
    imageToSave = double(imageToSave) / 255.0;
    
    % 保存图像到文件
    imwrite(imageToSave, fileName);
end

% 显示保存的图像
for i = 1:10
    fileName = fullfile(outputFolder, ['cifar10_image_label_' num2str(labels(i)) '_' num2str(i) '.png']);
    savedImage = imread(fileName);
    
    subplot(2, 5, i);
    imshow(savedImage);
    title(['Image ' num2str(i) ' - Label: ' num2str(labels(i))]);
end

Tips: 确保将 '/path/to/cifar-10-batches-mat/' 替换为你的 CIFAR-10 数据集的实际文件夹路径。注意是文件夹路径，不需要带文件的名称

②输出结果：

                                                                      matlab中显示的

              文件夹中显示的（不需要保存到文件夹中的，可以自行去掉保存图像部分的代码）

---

更多内容持续更新中，微信公众号同步，欢迎点赞+关注~

关注公众号：海绵科研