yolov5检测数据集标签数量

最新推荐文章于 2023-04-19 20:05:09 发布

qq_52217283

最新推荐文章于 2023-04-19 20:05:09 发布

阅读量955

点赞数

文章标签： python

本文链接：https://blog.csdn.net/qq_52217283/article/details/126131104

版权

为了检测转换是否有漏的,或者识别有漏的,加了每个标注小于数量的检测

# -*- coding:utf8 -*-
import os


def get_txt_list(path_, max_):
    result = []
    for files in os.listdir(path_):
        # print(files)
        with open(path_ + "\\" + files, "r") as f:
            length = f.read().split("\n")
            if len(length) <= max_:
                result.append(files)
    return result


if __name__ == '__main__':
    path = r"D:\Downloads\yolov5-6.1\runs\detect\exp2\labels"
    print(get_txt_list(path, 7))