代码如下：（自用）
import os
txt_path = r'E:\Downloads\mask_yolo_853_tra80\YOLOLabels\\' # 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))
版权声明：本文为qq_45168485原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。