23 lines
609 B
Python
23 lines
609 B
Python
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))
|