python - 从 ESA Sentinel-2 产品在 python 中加载 RGB 图像并使用 openCV 保存

Question

根据 ESA snap，对于 RGB 图像，我们应该将波段 4 放入红色 channel ，将波段 3 放入绿色 channel ，将波段 2 放入蓝色 channel 。我们如何使用 python 将这些波段读取到 numpy 数组中，以便我们可以进行任何我们想要的图像处理，然后将 RGB 图像保存在磁盘上？

from snappy import Product
from snappy import ProductIO
import numpy as np
import cv2

product = ProductIO.readProduct(path_to_product)

width = product.getSceneRasterWidth()
height = product.getSceneRasterHeight()

# Natural colors 
red = product.getBand('B4')
green = product.getBand('B3')
blue = product.getBand('B2')

例如这里是上述变​​量之一的类型(其他相同):

type(red)
# org.esa.snap.core.datamodel.Band

如何从这些数据中获取 numpy 数组，然后将它们作为 jpg 图像保存到磁盘？

Answer 1

#Read in channel's pixels    
red_pixels = np.zeros(width * height, np.float32)
red.readPixels(0, 0, width, height, red_pixels)

green_pixels = np.zeros(width * height, np.float32)
green.readPixels(0, 0, width, height, green_pixels)

blue_pixels = np.zeros(width * height, np.float32)
blue.readPixels(0, 0, width, height, blue_pixels)

#Reshape to image dimensions
red_pixels.shape =  height, width
green_pixels.shape =  height, width
blue_pixels.shape =  height, width

#Combine into a RGB image
rgb=np.zeros((height,width,3))
rgb[...,0] = red_pixels
rgb[...,1] = green_pixels
rgb[...,2] = blue_pixels

到目前为止，我们在具有浮点值的 numpy 数组中有一个 rgb 图像。为了以 jpg 图片的形式写入磁盘，我们首先裁剪较大的值以使图像更亮，然后将图像转换为 0-255 的整数值。

rgb2 = ((np.clip(rgb.copy(),0,1))*255).astype('uint8')
#Reverse Red-Blue Channels as open cv will reverse again upon writing image on disk
cv2.imwrite('image_name.jpg',rgb2[...,::-1])