import os

# file_path = '/home/admin-root/haotian/python哈汽安全帽识别/fire_smoke/remove/none.txt'

file_path = 'fire-8/remove/train.txt'

d = dict()

with open(file_path, 'r') as f:
    for line in f:
        d[line.strip('\n')] = 1

print(len(d))

# train_image_path = '/home/admin-root/haotian/python哈汽安全帽识别/fire_smoke/train/images'
# train_label_path = '/home/admin-root/haotian/python哈汽安全帽识别/fire_smoke/train/labels'

train_image_path = 'fire-8/train/images'
train_label_path = 'fire-8/train/labels'

# val_image_path = '/home/admin-root/haotian/python哈汽安全帽识别/fire_smoke/val/images'
# val_label_path = '/home/admin-root/haotian/python哈汽安全帽识别/fire_smoke/val/labels'
#
all_train_image = os.listdir(train_image_path)

for i in range(len(all_train_image)):
    # image = all_train_image[i].split('.')[0]
    image = all_train_image[i]
    if image in d:
        os.remove(os.path.join(train_image_path, all_train_image[i]))
        os.remove(os.path.join(train_label_path, image[:-4]+'.txt'))
        print(f'删除{image}')
#
# all_val_image = os.listdir(val_image_path)
#
# for i in range(len(all_val_image)):
#     image = all_val_image[i].split('.')[0]
#     if image in d:
#         os.remove(os.path.join(val_image_path, all_val_image[i]))
#         os.remove(os.path.join(val_label_path, image+'.txt'))

print('finished')