[发明专利]基于条件对抗神经网络的遥感影像自动匀色方法和装置

说明书

本发明提供了一种基于条件对抗神经网络的遥感影像自动匀色方法和装置，涉及图像处理的技术领域，包括：获取待匀色遥感影像和色彩模板影像，其中，色彩模板影像对应的区域范围包含待匀色遥感影像对应的区域范围；基于地理坐标，按照预设尺寸分别对待匀色遥感影像和色彩模板影像进行裁切，得到待匀色遥感影像块和色彩模板影像块；将待匀色遥感影像块和色彩模板影像块输入条件对抗神经网络模型，得到待匀色遥感影像块对应的目标遥感影像块，其中，目标遥感影像块为完成匀色的待匀色遥感影像块；对目标遥感影像块进行拼接，得到完成匀色的遥感影像，解决了现有的遥感影像自动匀色技术存在影像内部辐射精度不均匀或者较亮区域过饱和的技术问题。

技术领域

本发明涉及图像处理的技术领域，尤其是涉及一种基于条件对抗神经网络的遥感影像自动匀色方法和装置。

背景技术

现有的影像自动匀色处理算法都是基于模板影像或者标准色彩库进行的，通过统计模板影像和待匀色影像的均值、方差、直方图等信息并进行数学模型的同化处理，以达到待匀色影像的色彩在处理后能够更加贴近模板影像的色彩。然而，现有常规匀色方法大都是采用线性模型，适用于影像内部辐射较为均匀的影像，对于内部辐射不均匀的影像，传统的自动匀色方法通常得不到理想的效果；其次，遥感影像有些地物如白色房顶等本身是有一定的纹理信息的，传统匀色方法在统计计算和使用匀色系数后，会导致本身高亮的区域过饱和，丢失了遥感影像原本的纹理信息且不可逆。

针对上述问题，还未提出有效的解决方案。

发明内容

有鉴于此，本发明的目的在于提供一种基于条件对抗神经网络的遥感影像自动匀色方法和装置，以缓解了现有的遥感影像自动匀色技术存在影像内部辐射精度不均匀或者较亮区域过饱和的技术问题。

第一方面，本发明实施例提供了一种基于条件对抗神经网络的遥感影像自动匀色方法，包括：获取待匀色遥感影像和色彩模板影像，其中，所述色彩模板影像对应的区域范围包含所述待匀色遥感影像对应的区域范围；基于地理坐标，按照预设尺寸分别对所述待匀色遥感影像和色彩模板影像进行裁切，得到待匀色遥感影像块和色彩模板影像块；将所述待匀色遥感影像块和所述色彩模板影像块输入条件对抗神经网络模型，得到所述待匀色遥感影像块对应的目标遥感影像块，其中，所述目标遥感影像块为完成匀色的待匀色遥感影像块；对所述目标遥感影像块进行拼接，得到完成匀色的遥感影像。

进一步地，在基于地理坐标，按照预设尺寸分别对所述待匀色遥感影像和色彩模板影像进行裁切，得到待匀色遥感影像块和色彩模板影像块之前，所述方法还包括：判断所述待匀色遥感影像的地理坐标与所述色彩模板影像的地理坐标之间是否存在偏差；若是，则对所述待匀色遥感影像和所述色彩模板影像进行影像集合配准处理。

进一步地，所述条件对抗神经网络模型为基于Pix2PixHD构建的条件对抗神经网络模型。