分类图像数据集整理

简介

在准备深度学习数据集的时候，可能会出现所有类型图片都在一个文件夹里面，如mnist数据集。几万张图像一个个手动分离十分耗时，因此准备写个脚本将其剥离。
思路如下：
1、读所有图片的文件名
2、记录".jpg"前面的字符并以此创建新的文件夹，最后将该图像放置新建的文件夹中。

需要mnist数据集放最后了

输入

输出

代码

import os
import shutil

# 设置文件夹路径
folder_path = "F:/mnist/test"

# 获取文件夹中所有文件名
# test_39_1.jpg
file_names = os.listdir(folder_path)

# 遍历每个文件
for file_name in file_names:
    # 获取数字部分
    number = file_name.split("_")[-1].split(".")[0]

    # 创建数字文件夹
    if not os.path.exists(os.path.join(folder_path, number)):
        os.makedirs(os.path.join(folder_path, number))

    # 移动文件
    # shutil.move(A，B)  将A移动到B
    shutil.move(os.path.join(folder_path, file_name), os.path.join(folder_path, number, file_name))

print("完成！")

mnist数据集

sl12