计算机视觉中照片和标签对应检查问题

请叫我诗人

已于 2023-09-22 15:19:53 修改

阅读量129

点赞数

文章标签： java 服务器前端

于 2023-09-22 15:17:27 首次发布

本文链接：https://blog.csdn.net/REVERSE_FLASH01/article/details/133172347

版权

import os

def get_prefixes(folder, extension):
    # 遍历文件夹，获取所有文件的前缀
    prefixes = set()
    for filename in os.listdir(folder):
        if filename.endswith(extension):
            prefix = filename.split('.')[0]  # 前缀是文件名的第一部分
            prefixes.add(prefix)
    return prefixes

# 定义文件夹路径和文件扩展名
folder1 = "/path/to/folder1"  # 图片文件夹
folder2 = "/path/to/folder2"  # 标签文件夹
image_extension = ".jpg"  # 图片文件的扩展名
label_extension = ".txt"  # 标签文件的扩展名

# 获取两个文件夹中所有文件的前缀
image_prefixes = get_prefixes(folder1, image_extension)
label_prefixes = get_prefixes(folder2, label_extension)

# 找出缺失的图片文件和标签文件
missing_images = label_prefixes - image_prefixes
missing_labels = image_prefixes - label_prefixes

print("缺失的图片文件: ", missing_images)
print("缺失的标签文件: ", missing_labels)