这里可以指定其他拓展名文件,也可同时删除多种文件类型。
import os
def delete_images(directory):
for file in os.listdir(directory):
# 这里可以指定其他拓展名文件
# if file.endswith(".jpg") or file.endswith(".png") or file.endswith(".jpeg"):
if file.endswith(".png"):
os.remove(os.path.join(directory, file))
print(f"Deleted {file}")
# 指定目录
directory = "D:/A-Materials/数据集/INCR-seed"
# 调用函数删除图片文件
delete_images(directory)