代码如下：（自用）
import os
txt_path = r'E:Downloadsmask_yolo_853_tra80YOLOLabels\' # txt文件所在路径
class_num = 3 # 样本类别数
class_list = [i for i in range(class_num)]
class_num_list = [0 for i in range(class_num)]
labels_list = os.listdir(txt_path)
for i in labels_list:
file_path = os.path.join(txt_path, i)
file = open(file_path, 'r') # 打开文件
file_data = file.readlines() # 读取所有行
for every_row in file_data:
class_val = every_row.split(' ')[0]
class_ind = class_list.index(int(class_val))
class_num_list[class_ind] += 1
file.close()
# 输出每一类的数量以及总数
print(class_num_list)
print('total:', sum(class_num_list))
