从celeba选择50张戴眼镜的和50张不带眼镜的，将其copy到同一个文件夹下面

运行方式：   python load_sample_data.py

import os
import glob
import shutil
import sys

 
def cp_50_image():
    images = glob.glob('../abc_gan_data/celebA/*.jpg')
 
    sour_file = "../abc_gan_data/celebA"
 
    # 想把图像中带有那个属性的图像，就将该属性放到对应的target_attr
    target_attr = 'Eyeglasses'
 
    target_file = "../abc_gan_data/celebA_" + target_attr + "/"
    # 设置为True的时候 与target_attr相对的属性也会保存到对应的文件夹下
    need_inverse_attr = True
 
    # 确认一下目标文件是不是正确的
    # print("********tar_file******", target_file)
 
    # 之前每次都要创建一个文件夹 来存放移除的文件 现在改成自动创建 越自动化越好
    if not os.path.exists(target_file):
        os.makedirs(target_file)
    
    number_real_jpg = 0
    number_inverse_jpg = 0    

    with open('list_attr_celeba.txt', 'r') as open_file:
        lines = open_file.readlines()
        # 选择第二行也就是属性哪一行的所有属性 并且将其存放到all_attr之中
        all_attr = lines[1].strip('\n').split()
        # print("确认一下all_tags的类型和具体的属性值",type(all_tags),all_tags)
        for i in range(2, len(lines)):
            line = lines[i].strip('\n').split()
            #  存放的是文件名称
            # print(len(line))
            if number_inverse_jpg ==99 and number_real_jpg ==99:
                print("总共复制了%d张戴眼镜的和%d张不戴眼镜的"%(number_real_jpg, number_inverse_jpg))
                sys.exit()
            if int(line[all_attr.index(target_attr) + 1]) == 1:
                # tags[line[0]] = [1, 0]
                new_file = os.path.join(sour_file, line[0])
                # print("new_file 的名字是：",new_file)
                # 如果celebA中存在图像的话 就对图像移动操作
                exist = os.path.exists(new_file)
                if exist:
                    # shutil.move(new_file, target_file)
                    # 使用shutil.copy的时候 只是复制原始celeba中图像不会减少
                    if number_real_jpg < 99:
                        number_real_jpg = number_real_jpg + 1
                        shutil.copy(new_file, target_file)

            elif int(line[all_attr.index(target_attr) + 1]) == -1:
                if need_inverse_attr:
                    # 否则的话 可以将相反属性存放到里新的文件之中 自己动手做吧 这个没测试
                    new_file = os.path.join(sour_file, line[0])
                    # print("new_file 的名字是：",new_file)
                    exist = os.path.exists(new_file)
                    if exist:
                        if number_inverse_jpg < 99:
                            number_inverse_jpg = number_inverse_jpg + 1
                            shutil.copy(new_file, target_file)

            else:
                print("永远不可能执行 哦 哈哈哈哈！！！！！！！！！！")
 
 
cp_50_image()

 

实验结果：