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

@@北冥有鱼

已于 2022-10-20 16:39:02 修改

阅读量455

点赞数

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

于 2022-10-19 23:25:29 首次发布

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

版权

# coding: utf-8
"""
    将原始数据集进行划分成训练集、验证集和测试集
"""

import os
import glob
import random
import shutil

dataset_dir = os.path.join("..", "..", "Data", "cifar-10-png", "raw_test")
train_dir = os.path.join("..", "..", "Data", "train")
valid_dir = os.path.join("..", "..", "Data", "valid")
test_dir = os.path.join("..", "..", "Data", "test")

train_per = 0.8
valid_per = 0.1
test_per = 0.1


def makedir(new_dir):  # 目录是否存在，如果不存在则创建
    if not os.path.exists(new_dir):
        os.makedirs(new_dir)


if __name__ == '__main__':

    for root, dirs, files in os.walk(dataset_dir):  # dirs=0-9
        for sDir in dirs:
            imgs_list = glob.glob(os.path.join(root, sDir, '*.png'))  # 第0个文件夹里面的照片列表 第2个。。。。
            random.seed(666)