Abstract:Aiming at the problem of insignificant target morphological features, inaccurate detection and unclear boundary of small-target regions, and multitarget boundary overlap in multitarget complex image segmentation, combining the image segmentation mechanism of generative adversarial network with the feature enhancement method of nonlocal attention, a generative adversarial network fused with attention mechanism (AM-GAN) is proposed. The generative network in the model is composed of residual network and nonlocal attention module, which use the feature extraction and multiscale fusion mechanism of residual network, as well as feature enhancement and global information fusion ability of nonlocal spatial-channel dual attention to enhance the target features in the detection area and improve the continuity and clarity of the segmentation boundary. The adversarial network is composed of fully convolutional networks, which penalizes the loss of information in small-target regions by judging the authenticity of prediction and label segmentation and improves the detection ability of the generative adversarial model for small targets and the accuracy of multitarget segmentation. AM-GAN can use the GAN’s inherent mechanism that reconstruct and repair high-resolution image, as well as the ability of nonlocal attention global receptive field to strengthen detail features, automatically learn to focus on target structures of different shapes and sizes, highlight salient features useful for specific tasks, reduce the loss of image detail features, improve the accuracy of small-target detection, and optimize the segmentation boundary of multitargets. Taking medical MRI abdominal image segmentation as a verification experiment, multitargets such as liver, left/right kidney, and spleen are selected for segmentation and abnormal tissue detection. In the case of small and unbalanced sample datasets, the class pixels’ accuracy reaches 87.37%, the intersection over union is 92.42%, and the average Dice coefficient is 93%. Compared with other methods in the experiment, the segmentation precision and accuracy are greatly improved. It shows that the proposed method has good applicability for solving typical multitarget image segmentation problems such as small-target feature detection, boundary overlap, and offset deformation.

Image Semantic Segmentation Based on the GAN Auxiliary Network

See Clearer at Night: Towards Robust Nighttime Semantic Segmentation Through Day-Night Image Conversion

Semantic Segmentation by Improved Generative Adversarial Networks

Effective Image Restoration for Semantic Segmentation.

Multi-Source Fusion Image Semantic Segmentation Model of Generative Adversarial Networks Based on FCN

Self-Ensembling GAN for Cross-Domain Semantic Segmentation

A Deeply Supervised Semantic Segmentation Method Based on GAN

IMPROVING SEMANTIC SEGMENTATION OF HIGH-RESOLUTION REMOTE SENSING IMAGES USING WASSERSTEIN GENERATIVE ADVERSARIAL NETWORK

Incremental Focal Loss GANs.

UGC: Unified GAN Compression for Efficient Image-to-Image Translation

Local and Global GANs with Semantic-Aware Upsampling for Image Generation

Semi-supervised semantic segmentation based on Generative Adversarial Networks for remote sensing images

A Novel Generator With Auxiliary Branch for Improving GAN Performance

S Seg: A Three-Stage Unsupervised Foreground and Background Segmentation Network

Attention Guided Global Enhancement and Local Refinement Network for Semantic Segmentation

ComGAN: Unsupervised Disentanglement and Segmentation via Image Composition

Improving GAN Training via Feature Space Shrinkage

A Generative Adversarial Network Fused with Dual-Attention Mechanism and Its Application in Multitarget Image Fine Segmentation

An Enhanced Encoder-Decoder Network Architecture for Reducing Information Loss in Image Semantic Segmentation

ResGANet: Residual Group Attention Network for Medical Image Classification and Segmentation