move_repeat.py 927 Bytes
import os
import shutil

origin_dir = r'/data2/face_id/situ_other/train_norm_112_mix_add'
target_dir = r'/data2/face_id/situ_other/repeat'
result_file_path = r'clear_result1.txt'

result_file = open(result_file_path, 'r')
all_lines = result_file.readlines()

for i, line in enumerate(all_lines):
    print(i)
    arr = line.strip('\n').split(',')
    id_1, id_2 = arr[0], arr[1]
    # save_id_dir = os.path.join(target_dir, id_1)
    # if os.path.exists(save_id_dir) is False:
    #     os.mkdir(save_id_dir)
    
    # sub_id_1_dir = os.path.join(origin_dir, id_1)
    sub_id_2_dir = os.path.join(origin_dir, id_2)

    # save_id_1_dir = os.path.join(save_id_dir, id_1)
    save_id_2_dir = os.path.join(target_dir, id_2)

    # if os.path.exists(save_id_1_dir) is False:
    #     shutil.copytree(sub_id_1_dir, save_id_1_dir)
    # if os.path.exists(save_id_2_dir) is False:
    #     shutil.move(sub_id_2_dir, save_id_2_dir)