统计语义分割标签文件夹中的分割类别以及像素点数目

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

一、前言

提示：统计语义分割标签文件夹中的分割类别以及像素点数目：

---

二、代码

提示：代码如下（示例）：

import numpy as np
import os
from PIL import Image
print("Check datasets format, this may take a while.")
print("检查数据集格式是否符合要求，这可能需要一段时间。")
png_names = os.listdir("D:\\data\\MFD\\seg_out\\label")
print(len(png_names))
print(png_names[:5])
# -------------------------------------#
#   统计输出，各个像素点的值得个数
# -------------------------------------#
print("正在统计输出的图片每个像素点的数量")
classes_nums = np.zeros([256], np.int)
for png_name in png_names:
    png_file_name = os.path.join("D:\\data\\MFD\\seg_out\\label", png_name)

    png = np.array(Image.open(png_file_name), np.uint8)
    classes_nums += np.bincount(np.reshape(png, [-1]), minlength=256)

print("打印像素点的值与数量。")
print('-' * 37)
print("| %15s | %15s |" % ("Key", "Value"))
print('-' * 37)
for i in range(256):
    if classes_nums[i] > 0:
        print("| %15s | %15s |" % (str(i), str(classes_nums[i])))
        print('-' * 37)

2.结果示例

提示：结果：

---