yolo格式数据集标注框数量统计

最新推荐文章于 2024-11-02 15:21:28 发布

CV小老弟

最新推荐文章于 2024-11-02 15:21:28 发布

阅读量865

点赞数

分类专栏： yolo 文章标签： YOLO python 开发语言 Powered by 金山文档

本文链接：https://blog.csdn.net/qq_46381513/article/details/129403005

版权

yolo 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

该代码定义了一个名为`box_count_yolo`的函数，用于读取指定路径下的标签文件，统计每个类别的数量。它遍历文件夹，打开每个文件，解析内容并更新类别计数。在主函数中，提供数据集的标签路径和类别总数即可运行此脚本。

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

已封装成函数，直接在注释处修改成自己存放标签的路径, 数据集的类别数即可使用

from pathlib import Path


def box_count_yolo(path, num_classes):
    lis = [0 for _ in range(num_classes)]
    n = 0
    for f in Path.iterdir(Path(path)):
        with open(Path(f)) as file:
            temp = file.read()
            temp = temp.split()
            for num in temp:
                if len(str(num)) == 1:
                    lis[int(num)] += 1
            print(lis)
        n += 1
    print(f"--------{n}个文件处理完成--------")
    print(lis)


if __name__ == '__main__':
    box_count_yolo("data/0708data/labels/test", 7)  # 修改成自己存放标签的路径, 数据集的类别数

我的文件结构