比较image文件夹和label文件夹下的文件,删除在image文件夹中存在的文件但在label文件夹中不存在的文件。
import os
img_path = 'G:/cervical_cancer/data/roi/aug_roi_based_data_generation/train/'
label_path = 'G:/cervical_cancer/data/roi/aug_roi_based_data_generation/label/'
# 读取image文件文件列表
dirpath = os.listdir(img_path)
for image in dirpath:
# 读取文件名
image_name = image.split(".")[0]
if image.split(".")[1] == "jpg":
# 读取label文件中相同的文件名的json文件
json_path = os.path.join(label_path + image_name + ".json")
# 判断label文件夹中是否存在相同文件名的文件
if not os.path.exists(json_path):
image_path = os.path.join(img_path + image)
print(image_path)
os.remove(image_path)