eiseg打标签后批量将过程图像分类

from PIL import Image
import os

# 判断标识 此处将图像名称中带有字符‘pse’的保存至另一文件夹
str_find = 'pse'
# 文件夹路径
dir_root = "C:\\Users\\86188\\PycharmProjects\\unet-pytorch-main\\VOCdevkit\VOC2007\\JPEGImages\\label\\"
files = os.listdir(dir_root)
list_name = []
for file_name in files:

    img = Image.open(os.path.join(dir_root, file_name))
    a, b = os.path.splitext(file_name)
    nam_1 = a.find(str_find)

    if nam_1 != -1:
        # 另存的文件夹路径
        img.save("C:\\Users\\86188\\PycharmProjects\\unet-pytorch-main\\VOCdevkit\\VOC2007\\SegmentationClass\\" + a + '.' + b)




    

print('done')