Pytorch模型训练实用教程-代码解读（1）

VIP文章 @@北冥有鱼

已于 2022-10-20 18:11:25 修改

阅读量579

点赞数

分类专栏： pytorch模型训练实用教程文章标签： python 开发语言 pytorch

于 2022-10-18 21:37:47 首次发布

本文链接：https://blog.csdn.net/m0_60096200/article/details/127395819

版权

"""
    将cifar10的data_batch_12345 转换成 png格式的图片
    每个类别单独存放在一个文件夹，文件夹名称为0-9
"""
from imageio import imwrite
import numpy as np
import os
import pickle


base_dir = r"C:\Users\13004\Desktop\课程\研究生课程\PyTorch_Tutorial-master\PyTorch_Tutorial-master" #修改为当前Data 目录所在的绝对路径
data_dir = os.path.join(base_dir, "Data", "cifar-10-batches-py")
train_o_dir = os.path.join( base_dir, "Data", "cifar-10-png", "raw_train")
test_o_dir = os.path.join( base_dir, "Data", "cifar-10-png", "raw_test")

Train = False   # 不解压训练集，仅解压测试集

# 解压缩，返回解压后的字典
def unpickle(file):
    with open(file, 'rb') as fo:
        dict_ = pickle.load(fo, encoding='bytes')
    return dict_

def my_mkdir(my_dir):
    if not os.path.isdir(my_dir):
        os.makedirs(my_dir)


# 生成训练集图片，
if __name__ == '__main__':
    if