CIFAR10数据集预处理的两种方式

软7

已于 2025-02-21 11:39:50 修改

阅读量589

点赞数 11

文章标签： python

于 2025-02-21 11:36:20 首次发布

本文链接：https://blog.csdn.net/qq_55856745/article/details/145773222

版权

（1）方法1：从CIFAR-10数据集中提取图像，并将它们按类别保存为PNG格式的文件。

目标目录结构如图：

import numpy as np

import pickle

import os

from torchvision import datasets

from imageio import imwrite

# 数据集放置路径 cifar10/cifar-10-batches-py

data_save_pth = "./"

train_pth = os.path.join(data_save_pth, "train")

test_pth = os.path.join(data_save_pth, "test")

# 创建必要的目录

def create_dir(path):

if not os.path.exists(path):

os.makedirs(path)

create_dir(train_pth)

create_dir(test_pth)

# 解压路径

data_dir = os.path.join(data_save_pth, "cifar-10-batches-py")

# 数据集下载</