SECBNet: Semantic Segmentation Enhanced Color Balance Network for Optical Satellite Images

Ziyi Chen,Hanhuang Chen,Lujuan Gao,Dilong Li,Cheng Wang,Linlin Xu,Somayeh Mollaee,Jonathan Li
DOI: https://doi.org/10.1109/tgrs.2024.3509735
IF: 8.2
2024-01-01
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Earth observation satellites can capture optical images under different temporal, climatic conditions, and platforms exhibit substantial differences in color and brightness, leading to poor visual experiences when synthesizing large-area optical satellite images. The related issue of color balancing has attracted considerable attention from researchers, yet challenges such as a lack of research data and sensitivity to model parameters persist. To address these problems, this paper publishes a publicly open dataset and presents a Semantic Segmentation Enhanced Color Balance Network (SECBNet). Firstly, to mitigate the scarcity of research data, we develop a publicly available remote sensing image color balance dataset, ZHCBI (Zhu Hai Color Balance Image), to support related research activities. Secondly, to improve semantic consistency between the color-balanced images and the target images, we design a dual-branch U-Net architecture guided by segmentation results and propose a novel segmentation feature loss function. Lastly, to address issues of seams and unnatural transitions between blocks in segmented processing, we introduce a post-processing module based on weighted averaging. We conducted comparative experiments and analyses with existing mainstream color balancing algorithms on the ZHCBI dataset. The results demonstrate that our proposed method achieves state-of-the-art color balancing quality, with significant improvement in visual effects and a higher peak signal-to-noise ratio (23.64 dB) compared to other mainstream methods.
What problem does this paper attempt to address?