YOLO标签数量统计

YOLO标签数量统计源代码

提示：确保标签文件已经转化成txt格式，才可以进行统计

#计算txt标签格式每个类别的数量
import os
from tqdm import tqdm
def get_every_class_num(txt_path):
    # 需修改，根据自己的类别，注意一一对应
    class_categories=['0', '1', '2'