43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
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')
|