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

最新推荐文章于 2024-02-16 19:15:43 发布

ye1256

最新推荐文章于 2024-02-16 19:15:43 发布

阅读量288

点赞数 1

分类专栏：深度学习文章标签： python 机器学习 numpy

本文链接：https://blog.csdn.net/ye1256/article/details/127431313

版权

深度学习专栏收录该内容

11 篇文章 2 订阅

订阅专栏

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][1])
    num1=+1

# print(classes_data[0][1])
# print(classes_data[1][1])
# print(classes_data[2][1])
# print(classes_data[3][1])

fig = plt.figure()
fig.canvas.set_window_title('训练数据0类别统计')
# plt.rcParams["font.sans-serif"] = ['SimHei']
plt.rcParams['font.sans-serif']=['SimHei'] #显示中文标签
plt.rcParams['axes.unicode_minus']=False   #解决负号“-”显示为方块的问题

plt.title("训练数据类别统计")
plt.xlabel("类别")
plt.ylabel("数量")
for i in range(len(classes_data)):
    plt.bar(classes_data[i][0], classes_data[i][1])
    # print (plt.bar(classes_data[i][0], classes_data[i][1]))


num_list = [classes_data[0][1],classes_data[1][1],classes_data[2][1],classes_data[3][1]]#分解
index=np.arange(len(num_list));
# num_list1 = [248.50326,56.595,17.913,283.33]#真实
for a,b in zip(index,num_list):   #柱子上的数字显示
    plt.text(a,b,'%.2f'%b,ha='center',va='bottom',fontsize=7);
plt.show()

结果：