from PIL import Image
import numpy as np
import os
from osgeo import gdal


mask_dir = r"E:\datasets\wlk_right_448\mask"  # 修改为你的mask文件夹路径

all_labels = set()

for file in os.listdir(mask_dir):
    if file.lower().endswith('.tif'):
        tif_path = os.path.join(mask_dir, file)
        dataset = gdal.Open(tif_path)
        if dataset is None:
            print(f"无法打开: {tif_path}")
            continue
        band = dataset.ReadAsArray()
        unique = np.unique(band)
        all_labels.update(unique)

print("所有mask中出现过的标签数字：", sorted(all_labels))