python读取多通道tif文件并显示rgb图像

AAAmazing

已于 2022-07-05 17:55:10 修改

阅读量6.2k

点赞数 1

文章标签： python 开发语言后端

于 2022-03-05 11:47:09 首次发布

本文链接：https://blog.csdn.net/AAAmazing/article/details/123292252

版权

from osgeo import gdal
import numpy as np
import matplotlib.pyplot as plt

data = gdal.Open('zxf.tif')   # 读取tif文件
num_bands = data.RasterCount     # 获取波段数
print(num_bands)
tmp_img = data.ReadAsArray()      #将数据转为数组
img_rgb = tmp_img.transpose(1, 2, 0)     #由波段、行、列——>行、列、波段

img_rgb = np.array(img_rgb, dtype=np.uint8)   #设置数据类型，np.unit8可修改
r = img_rgb[:, :, 0]
g = img_rgb[:, :, 3]
b = img_rgb[:, :, 2]
img_rgb = np.dstack((r, g, b))    # 波段组合

# plt.imshow(img_rgb)
# plt.show()
# 通过调用plt.axis（“ off”），可以删除编号的轴
plt.axis("off")
plt.imshow(img_rgb)
plt.show()