34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
import os
|
|
import numpy as np
|
|
from PIL import Image
|
|
from osgeo import gdal
|
|
|
|
# 输入和输出文件夹
|
|
src_dir = r"E:\datasets\WLKdata_1111\WLKdataset\images_LS"
|
|
dst_dir = r"E:\datasets\WLKdata_1111\WLKdataset\images_LS_jpg"
|
|
os.makedirs(dst_dir, exist_ok=True)
|
|
|
|
for file in os.listdir(src_dir):
|
|
if file.lower().endswith('.tif'):
|
|
tif_path = os.path.join(src_dir, file)
|
|
dataset = gdal.Open(tif_path)
|
|
if dataset is None:
|
|
print(f"无法打开: {tif_path}")
|
|
continue
|
|
|
|
# 读取所有波段
|
|
bands = []
|
|
for i in range(1, dataset.RasterCount + 1):
|
|
band = dataset.GetRasterBand(i).ReadAsArray()
|
|
bands.append(band)
|
|
img = np.stack(bands, axis=-1) if len(bands) > 1 else bands[0]
|
|
|
|
# 转换为uint8
|
|
img = img.astype(np.uint8)
|
|
|
|
jpg_path = os.path.join(dst_dir, os.path.splitext(file)[0] + ".jpg")
|
|
Image.fromarray(img).save(jpg_path, quality=95)
|
|
print(f"已保存: {jpg_path}")
|
|
|
|
print("全部转换完成！")
|