import os

# 文件夹路径

image_folder = input("请输入图片路径:") # 图片文件夹

text_folder = input("请输入txt路径:") # 文本文件夹

# 获取所有图片和文本文件的名字（不带扩展名）

image_files = {os.path.splitext(f)[0] for f in os.listdir(image_folder) if f.lower().endswith(('.jpg', '.png', '.jpeg'))}

text_files = {os.path.splitext(f)[0] for f in os.listdir(text_folder) if f.lower().endswith('.txt')}

# 找出不匹配的文件

images_to_delete = image_files - text_files # 没有对应 txt 的图片

texts_to_delete = text_files - image_files # 没有对应图片的 txt（如果需要删除）

# 删除多余的图片

for img in images_to_delete:

img_path = os.path.join(image_folder, img + ".jpg") # 以 .jpg 为例

if not os.path.exists(img_path): # 兼容 .png/.jpeg

img_path = os.path.join(image_folder, img + ".png")

if os.path.exists(img_path):

os.remove(img_path)

print(f"Deleted image: {img_path}")

# 删除多余的文本（可选）

for txt in texts_to_delete:

txt_path = os.path.join(text_folder, txt + ".txt")

if os.path.exists(txt_path):

os.remove(txt_path)

print(f"Deleted text file: {txt_path}")

print("清理完成！")

代码解析

1. 获取用户输入的文件夹路径

image_folder = input("请输入图片路径:") # 图片文件夹 text_folder = input("请输入txt路径:") # 文本文件夹

• input() 让用户输入 存放图片的文件夹路径 和 存放文本的文件夹路径。
• image_folder 用于存放 图片（.jpg、.png、.jpeg）。
• text_folder 用于存放 文本文件（.txt）。

2. 获取文件夹中的文件名（不含扩展名）

image_files = {os.path.splitext(f)[0] for f in os.listdir(image_folder) if f.lower().endswith(('.jpg', '.png', '.jpeg'))} text_files = {os.path.splitext(f)[0] for f in os.listdir(text_folder) if f.lower().endswith('.txt')}

• os.listdir(image_folder) 获取 图片文件夹 下的所有文件名。
• os.path.splitext(f)[0] 获取 文件名（去掉扩展名）。
• if f.lower().endswith(('.jpg', '.png', '.jpeg')) 只保留 图片文件。
• 以上步骤得到 image_files（图片文件名集合） 和 text_files（文本文件名集合）。

图片文件夹： image1.jpg image2.png image3.jpeg 文本文件夹： image1.txt image2.txt

得到：

image_files = {"image1", "image2", "image3"} text_files = {"image1", "image2"}

3. 找出没有匹配的文件

images_to_delete = image_files - text_files # 没有对应 txt 的图片 texts_to_delete = text_files - image_files # 没有对应图片的 txt（如果需要删除）

• image_files - text_files：找出没有对应 .txt 文件的图片。
• text_files - image_files：找出没有对应图片的 .txt 文件。

按照上面的示例：

images_to_delete = {"image3"} # 没有对应 image3.txt texts_to_delete = set() # 没有多余的 txt 文件

4. 删除没有匹配的图片

for img in images_to_delete: img_path = os.path.join(image_folder, img + ".jpg") # 以 .jpg 为例 if not os.path.exists(img_path): # 兼容 .png/.jpeg img_path = os.path.join(image_folder, img + ".png") if os.path.exists(img_path): os.remove(img_path) print(f"Deleted image: {img_path}")

• 遍历 images_to_delete，找到 没有对应文本文件的图片。
• 依次尝试 .jpg、.png 等扩展名，检查文件是否存在。
• 如果找到对应的图片文件，就 删除 并打印 "Deleted image: 路径"。

5. 删除没有匹配的文本文件

for txt in texts_to_delete: txt_path = os.path.join(text_folder, txt + ".txt") if os.path.exists(txt_path): os.remove(txt_path) print(f"Deleted text file: {txt_path}")

• 遍历 texts_to_delete，找到 没有对应图片的 .txt 文件 并删除。
• （可选） 如果不想删除多余的 .txt，可以删除 for txt in texts_to_delete: 这一部分。

6. 结束程序

print("清理完成！")

• 提示用户清理完成。

代码执行示例

文件夹内容（执行前）

/images/ image1.jpg image2.png image3.jpeg image4.jpg (无对应 txt) /texts/ image1.txt image2.txt image5.txt (无对应图片)

代码执行后

删除 image4.jpg（无对应 txt） 和 image5.txt（无对应图片）。

删除的文件列表

Deleted image: /images/image4.jpg Deleted text file: /texts/image5.txt 清理完成！

文件夹最终内容

/images/ image1.jpg image2.png image3.jpeg /texts/ image1.txt image2.txt

总结

功能

• 检查图片和文本文件是否匹配。
• 删除没有对应文本的图片。
• （可选）删除没有对应图片的文本文件。

适用场景

• 适用于 AI 训练数据清理（确保图片和标注文件匹配）。