SRGAN生成卫星超分辨率图像颜色与拼接异常问题咨询

我基于SRGAN训练脚本开展卫星图像超分辨率任务，仅修改了输入图像、尺寸及图像保存逻辑，当前遇到两类问题：

• 生成图像状态异常，RGB像素值集中在1附近，其余颜色像素值处于0.x区间，但输入数据集的取值范围为0-0.6
• 已将所有图像分割为32×32尺寸，期望在ArcGIS中加载时能拼接排列，实际却呈现叠加显示的情况

我测试过预训练ImageNet模型与从头搭建的VGG模型，相关代码如下：

加载低分辨率图像代码 #

s2_image = []
def load_s2_images(s2):
    for file in os.listdir(s2):
        if file.endswith('.TIF'):
            file_path = os.path.join(s2, file)

            dataset = gdal.Open(file_path, gdalconst.GA_ReadOnly)
            red = dataset.GetRasterBand(4).ReadAsArray()
            green = dataset.GetRasterBand(3).ReadAsArray()
            blue = dataset.GetRasterBand(2).ReadAsArray()
            # coastal = dataset.GetRasterBand(1).ReadAsArray()
            # nir = dataset.GetRasterBand(8).ReadAsArray()
            # redEg = dataset.GetRasterBand(5).ReadAsArray()

            s2_nparray = np.stack((red, green, blue), axis=-1)#, coastal, nir, redEg), axis=-1)

            s2_image.append(s2_nparray)
            
    return s2_image
            
s2_image_arrays = load_s2_images(s2)

保存图像代码 #

def save_generated_images(epoch, generator_model, s2_test, output_dir):
    generated_sr_images = generator_model.predict(s2_test)
    for ni, image in enumerate(generated_sr_images):
        plt.imsave(f"{output_dir}/epoch_{epoch}_image_{i}.png", image)

内容的提问来源于stack exchange，提问作者User2010