代码:
import os
coco_train_label_path = r"C:\Users\9ling\Desktop\luggage_dataset\coco_2014\labels\val2014"
coco_train_image_path = r"C:\Users\9ling\Desktop\luggage_dataset\coco_2014\val2014"
new_coco_train_image_path = r"C:\Users\9ling\Desktop\luggage_dataset\val_luggage"
for item in os.listdir(coco_train_label_path):
txt_path = os.path.join(coco_train_label_path, item)
with open(txt_path, 'r') as f:
data = f.readlines()
# print(data)
for j in data:
if j.split(" ")[0] == "28":
print("28"*10)
# print(item)
new_path = os.path.join(new_coco_train_image_path, item.split('.')[0]+'.jpg')
image_path = os.path.join(coco_train_image_path, item.split('.')[0]+'.jpg')
# break
with open(image_path, 'rb') as f1:
contents = f1.read()
# new_path = os.path.join(new_coco_train_image_path, item.split('.')[0], 'jpg')
with open(new_path, 'wb') as f2:
f2.write(contents)
print("***********")
break