import os
from osgeo import gdal
import numpy as np
from PIL import Image
# 提取432三波段
from spectral import *
# 输入文件夹路径
def get_img(dataset_img):
width = dataset_img.RasterXSize # 获取行列数
height = dataset_img.RasterYSize
bands = dataset_img.RasterCount # 获取波段数
# print("行数为:", height)
# print("列数为:", width)
# print("波段数为:", bands)
proj = dataset_img.GetProjection() # 获取投影信息
# print("投影信息", proj)
geotrans = dataset_img.GetGeoTransform() # 获取仿射矩阵信息
img = dataset_img.ReadAsArray(0, 0, width, height)
new_img = np.transpose(img, (1, 2, 0))
return new_img
def tiftopng(input_folder,output_folder,band=[9,3,2]):
# 确保输出文件夹存在
if not os.path.exists(output_folder):
os.makedirs(output_folder)
# 遍历输入文件夹中的.tif文件
for file_name in os.listdir(input_folder):
if file_name.endswith('.tif'):
# 构建输入文件的完整路径
input_file = os.path.join(input_folder, file_name)
# 打开.tif文件
image = get_img(gdal.Open(input_file))
# 创建输出文件的完整路径
output_file = os.path.join(output_folder, file_name.replace('.tif', '.png'))
spectral.settings.WX_GL_DEPTH_SIZE = 16
# save_rgb(output_file, image, bands=[29, 3, 2], format='png')
save_rgb(output_file, image, bands=band, format='png')
input_folder = r'F:\datasets\ZY\ZY_S2_TEST\540\Z'
# 输出文件夹路径
output_folder = r'F:\datasets\ZY\ZY_S2_TEST\540\Z_png'
tiftopng(output_folder,input_folder)
if __name__ == '__main__':
root = r'F:\datasets\ZY\ZY_S2_TEST\\540'
x_path = root + '\X'
y_path = root + '\Y'
z_path = root + '\Z'
out_root = 'F:\datasets\ZY\ZY_S2_TEST\\540'
out_x = out_root + '\X_png'
out_y = out_root + '\Y_png'
out_z = out_root + '\Z_png'
tiftopng(y_path, out_y,band=[9,3,2])
tiftopng(z_path, out_z,band=[29,3,2])
tiftopng(x_path, out_x,band=[9,3,2])
python tif转png
于 2024-04-01 21:14:55 首次发布