CMAE-Unet: A Study on a U-Net-Based Model for Semantic Segmentation of Unripe Tomato Images
TL;DR
This work proposes CMAE-UNet, a camouflage suppression and edge enhancement model based on the UNet architecture that robustly withstands field interference, providing strong technological support for automated tomato detection, intelligent harvesting, and growth monitoring.
Abstract
In natural scenes, the high resemblance between unripe tomatoes and foliage, combined with severe occlusion, challenges current semantic segmentation models, causing poor accuracy and indistinct boundaries. To overcome this, we propose CMAE-UNet, a camouflage suppression and edge enhancement model based on the UNet architecture. Specifically, a Global-Local Integrated Spatial Attention (GLISA) encoder merges dual-branch dilated convolutions, residual structures, and an Efficient Multi-scale Attention mechanism to expand receptive fields and highlight targets in complex backgrounds. Furthermore, a Frequency-Domain Feature Enhancement (FFE) module leverages the Fast Fourier Transform to separate and adaptively enhance distinct frequency components, effectively mitigating camouflage interference. Additionally, a Directional Edge Enhancement (DEE) module uses three-directional learnable convolutions and spatial attention to sharpen indistinct target contours. Evaluated on a custom Tomato dataset encompassing five complex scenarios, CMAE-UNet outperforms 12 prominent methods in mIoU, Dice, and Sen metrics, yielding smoother and more precise segmentation boundaries. The model robustly withstands field interference, providing strong technological support for automated tomato detection, intelligent harvesting, and growth monitoring.