检查图片与标签 img和label是否一一对应解决coco128数据集不对应

373023820

已于 2023-10-11 15:21:13 修改

阅读量752

点赞数 2

文章标签：人工智能

于 2023-10-11 15:20:29 首次发布

本文链接：https://blog.csdn.net/qq_53821866/article/details/133771208

版权

本文介绍了一个Python程序，它遍历指定的图像文件夹和标签文件夹，检查每个图像文件是否都有与其后缀不同的对应标签文件。若标签文件不存在，程序会输出提示。

摘要由CSDN通过智能技术生成

import os

image_dir = 'coco128/images/train2017'  # 图像文件夹路径
label_dir = 'coco128/labels/train2017'  # 标签文件夹路径

# 获取图像文件夹中的所有文件名
image_files = [f for f in os.listdir(image_dir) if f.endswith('.jpg')]

# 获取标签文件夹中的所有文件名
label_files = [f for f in os.listdir(label_dir) if f.endswith('.txt')]

# 检查每个图像文件是否有对应的标签文件
for image_file in image_files:
    # 组合图像文件的路径和标签文件的路径
    image_file_path = os.path.join(image_dir, image_file)
    label_file_path = os.path.join(label_dir, image_file.replace('.jpg', '.txt'))

    if not os.path.exists(label_file_path):
        print(f'标签文件 {label_file_path} 不存在与图像文件 {image_file_path} 对应')

需要将 image_dir 和 label_dir 替换为你实际的图像文件夹和标签文件夹的路径。该程序假设图像文件以 .jpg 结尾，标签文件以 .txt 结尾，并且它们的文件名是一样的，只是后缀不同（例如图像文件为 image1.jpg，则对应的标签文件应该是 image1.txt）。

程序遍历图像文件夹中的每个图像文件，组合图像文件路径和标签文件路径，并检查标签文件是否存在。如果标签文件不存在，则打印一条消息指示图像文件和对应的标签文件不匹配。