根据txt文档解析图片分类

txt文档里面已经对图片文件进行了分类

代码

import os
import shutil

# 读入分类的标签txt文件
label_file = open("data/Datasets/picture_list.txt", 'r',encoding="utf-8")
# 原始文件的根目录
input_path = "data/Datasets/jpg"
# 保存文件的根目录
output_path = "data/Datasets"
# 分类数组
lables = ["0","1"]

# 逐行读入标签文件
data = label_file.readlines()
# 计数
i = 1
# 遍历数据
for line in data:
    # 根据txt每条显示信息的结果拆分成str
    str1 = line.split("/")
    str2=str1[1].split(" ")
    # 第一个是文件名
    file=str1[0]
    file_name = str2[0]
    # 第二个是标签类别，并去除最后的换行字符
    file_label = str2[1].strip()
    # 原始文件的路径
    old_file_path = os.path.join(input_path, file_name)
    # 新文件路径
    new_file_path = "data"

    # 如果文件名中有0字符，将其保存至0文件夹下的对应标签文件夹中
    if "0" in file_label:
        new_file_path = os.path.join(output_path, lables[int(file_label)])
    # 如果文件名中有1字符，将其保存至1文件夹下的对应标签文件夹中
    elif "1" in file_label:
        new_file_path = os.path.join(output_path, lables[int(file_label)])

    # 如果路径不存在，则创建
    if not os.path.exists(new_file_path):
        os.makedirs(new_file_path)

    # 新文件位置
    new_file_path = os.path.join(new_file_path, file_name)
    # print( str(i) +" " + "正在将 " + old_file_path + " 复制到 " + new_file_path)
    # 复制文件
    shutil.copyfile(old_file_path, new_file_path)

    i = i + 1
# 完成提示
print("完成")

结果

代码读取后自动执行，将jpg下面的图片按照txt文档中的分类标签顺利进行分类


参考了一位博主的代码并根据自己的需求进行了修改，不过我不小心找不到他的原博了sorry