np.unique函数获取mask图片中不同像素数

np.unique函数获取mask图片中不同像素数

@Data 2022/12/9

代码：

import os
import numpy as np
from PIL import Image


def unique_values(root):
    imgs = os.listdir(root)
    concat_unique = np.empty(1)
    for imgpath in imgs:
        img = np.asarray(Image.open(os.path.join(root, imgpath)))
        # 得到像素中的不同数
        unique = np.unique(img)
        # 对其进行拼接
        concat_unique = np.concatenate([concat_unique, unique])
    # 对拼接后的图片进行再次求不同像素，即全部文件中不同像素数，排序后返回
    return list(sorted(np.unique(concat_unique)))


if __name__ == '__main__':
    root = r'./out'
    unique = unique_values(root)
    print(unique)

输出：

>>[0.0 15.0]