19 lines
478 B
Python
19 lines
478 B
Python
from PIL import Image
|
||
import numpy as np
|
||
import os
|
||
from osgeo import gdal
|
||
|
||
|
||
mask_dir = r"E:\RSdata\wlk_right_448\mask_png_ori" # 修改为你的mask文件夹路径
|
||
|
||
all_labels = set()
|
||
|
||
for file in os.listdir(mask_dir):
|
||
if file.lower().endswith('.png'):
|
||
img = Image.open(os.path.join(mask_dir, file))
|
||
img = np.array(img)
|
||
unique = np.unique(img)
|
||
all_labels.update(unique)
|
||
|
||
print("所有mask中出现过的标签数字：", sorted(all_labels))
|