根据label中的txt文件统计不同class的数量，代码实现

最新推荐文章于 2024-11-04 22:03:20 发布

ye1256

最新推荐文章于 2024-11-04 22:03:20 发布

阅读量346

点赞数 1

分类专栏：深度学习

本文链接：https://blog.youkuaiyun.com/ye1256/article/details/127431313

版权

Python 数据集类别统计 Matplotlib 类别可视化

关键词由优快云通过智能技术生成

import os
import matplotlib.pyplot as plt
import numpy as np

folder_path = r"D:\YWJ\5_zsj_dataset\3500_division\val\labels"  # 存放Labels路径
classes_file = "classes.txt"  # 存放类别文件路径

classes_data = []
file_list = os.listdir(folder_path)

# print(file_list)

for classes in open(classes_file, 'r').readlines():
    dr = [classes.replace("\n", ""),0]
    classes_data.append(dr)
    # print(classes_data)

for item in file_list:
    if item.split('.')[-1] == "txt" and item != "classes.txt":
        for line in open(folder_path + "/" + item, 'r').readlines():
            for i in range(len(classes_data)):
                if line.split(' ')[0] == str(i):
                    classes_data[i][1] += 1
                    # print(max(classes_data[i][1]))

for num1 in range(0,4):
    # num1=0
    print(classes_data[num1][