检查文件夹里面的jpg图片是否有相对应的xml标注文件，反之亦然。

原创于 2024-05-02 22:27:49 发布 · 98 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#xml #java #前端

Python脚本检测文件夹中jpg与xml文件的对应关系，缺失的文件将被删除。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

检查文件夹里面的jpg图片是否有相对应的xml标注文件，反之亦然。


```python
import os  
  
def remove_unpaired_files(directory):  
    jpg_files = set()  
    xml_files = set()  
      
    # 遍历目录，收集所有的jpg和xml文件名  
    for filename in os.listdir(directory):  
        if filename.endswith('.jpg'):  
            jpg_files.add(filename[:-4])  # 移除.jpg扩展名以进行匹配  
        elif filename.endswith('.xml'):  
            xml_files.add(filename[:-4])  # 移除.xml扩展名以进行匹配  
      
    # 找出没有对应xml的jpg文件  
    unpaired_jpgs = jpg_files - xml_files  
    # 找出没有对应jpg的xml文件  
    unpaired_xmls = xml_files - jpg_files  
      
    # 删除没有对应文件的jpg和xml  
    for unpaired_jpg in unpaired_jpgs:  
        os.remove(os.path.join(directory, unpaired_jpg + '.jpg'))  
    for unpaired_xml in unpaired_xmls:  
        os.remove(os.path.join(directory, unpaired_xml + '.xml'))  
  
# 示例使用  
directory = '/path/to/your/directory'  # 替换为你的目录路径  
remove_unpaired_files(directory)