Portrait semantic segmentation is one of the important research contents in the field of computer vision
but the existing portrait semantic segmentation methods are liable to ignore the small size portraits in multi-person portrait images. At the same time
the segmentation results are prone to the phenomenon of mutual adhesion between multiple portraits. Moreover
the phenomenon of mutual occlusion between portraits in the image easily leads to poor segmentation accuracy of portrait edges. Based on the above problems
a semantic segmentation method for multiple portraits with fused label semantics is propose
where multiple labels are assigned to multiple portraits in an image
and semantic labels are embedded as inputs to the encoder at the same time
and the semantic labels and the image feature representations are correlated using the cross-modal cross-attention module
and the semantically fused feature representations are obtained as outputs of the encoder at each layer of the model. The HRF attention module is proposed to generate multiple hypotheses for image based on target detection algorithm for feature extraction separately. The network is trained and tested on Supervisely augmented dataset. The experimental results show that the algorithmic model achieves 95.94%
94.60%
and 96.02% accuracy on the three evaluation metrics of PA
MIoU
and Dice
respectively
and has higher segmentation accuracy than the semantic segmentation models U-net
PSPNet
Deeplab v3+
PortraitNet
and Swin Unet.
关键词
Keywords
references
王欣. 基于深度学习的人像分割方法研究[D]. 咸阳: 西北农业科技大学, 2022.
KUAI Y, WANG B, WU Y L, et al . Urban vegetation classification based on multi-scale feature perception network for UAV images[J ] . Journal of Geo-Information Science, 2022, 24(5): 962-980.
RONNEBERGER O, FISCHER P, BROX T. U-net: convolutional networks for biomedical image segmentation[C] //Medical Image Computing and Computer-assisted Intervention–MICCAI 2015: 18th International Conference. Munich: Springer International Publishing, 2015: 234-241.
WEI Y, XIA W, LIN M, et al . HCP: a flexible CNN framework for multi-label image classification[J ] . IEEE, 2015, 38(9): 1901-1907.
AZAD R, ASADI-AGHBOLAGHI M, FATHY M, et al . Attention DeepLabv3+: multi-level context attention mechanism for skin lesion segmentation[C ] //European Conference on Computer Vision (ECCV) . Glasgow: Springer International Publishing, 2020: 251-266.
ZHAO W P, CHEN Y, XIANG S, et al . Research on image s emantic segmentation algorithm based on improved DeepLabv3+[J ] . Journal of System Simulation, 2023, 35(11): 2333-2344.
YANG Y, WAN W, HUANG S, et al . RADCU-Net: residual attention and dual-supervision cascaded U-Net for retinal blood vessel segmentation[J ] . International Journal of Machine Learning and Cybernetics, 2023, 14: 1605-1620.
FU J, LIU J, TIAN H, et al . Dual attention network for scene segmentation[C ] //Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Long Beach: IEEE, 2019: 3146-3154.
AMER A, LAMBROU T, YE X. MDA-unet: a multi-scale dilated attention U-net for medical image segmentation[J]. Applied Sciences, 2022, 12(7): 3676.
雷俊婷. 基于多尺度特征增强的多标签图像分类研究[D]. 西安: 西北大学, 2023.
ZHANG S J, PENG Z, LI H. SAU-Net: medical image seg- mentation method based on U-Net and self-attention[J]. Acta Elect Ronica Sinica, 2022, 50(10): 2433-2442.
WU W, LIU H, LI L, et al . Application of local fully convolutional neural network combined with YOLO v5 algorithm in small target detection of remote sensing image[J ] . PloS One, 2021, 16(10): e0259283.
CHEN L C, ZHU Y, PAPANDREOU G, et al . Encoder-decoder with atrous separable convolution for semantic image segmentation[C ] //Proceedings of the European Conference on Computer Vision (ECCV) . Munich: Springer International Publishing, 2018: 801-818.
ZHANG Y, TIAN Y, KONG Y, et al . Residual dense network for image super-resolution[C ] //Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Calgary: IEEE, 2018: 2472-2481.
DU M M, SIMA H F. A-LinkNet: semantic segmentation net-work based on attention and spatial information fusion[J]. Chinese Journal of Liquid Crystals and Displays, 2022, 37(9): 1199-1208.
WANG X Y, WANG C R, ZHANG J F, et al . Multimodal sentiment analysis based on cross-modal cross-attention network [J ] . Journal of Guangxi Normal University (Natural Science Edition) , 2024, 42(2) : 84-930.
SUN B, JIANG T, JIA L, et al . Multimodal sentiment analysis based on cross-modal joint-encoding[J ] . Computer Engineering and Applications, 2024, 60(18): 208-216.
CHEN Q, GE T, XU Y, et al . Semantic human matting[C ] //Proceedings of the 26th ACM International Conference on Multi-media. New York: ACM, 2018: 618-626.
HUANG Y J, SHI Z F, WANG Z Q, et al . Improved U-net based on mixed loss function for liver medical image segmentation[J ] . Laser Optoelectronics Progress, 2020, 57(22): 74-83.
ZHAO H, SHI J, QI X, et al . Pyramid scene parsing network[C ] //Proceedings of the IEEE Conference On Computer Vision and Pattern Recognition. Kyoto: IEEE, 2017: 2881-2890.
CHEN L, ZHU Y, PAPANDREOU G, et al . Encoder-decoder with atrous separable convolution for semantic image segmentation[C ] //Proceedings of the European Conference on Computer Vision (ECCV) . Munich: Springer International Publishing, 2018: 801-818.
ZHANG S H, DONG X, LI H, et al . PortraitNet: realtime portrait segmentation network for mobile device[J ] . Computers Graphics, 2019, 80: 104-113.
CAO H, WANG Y, CHEN J, et al . Swin-unet: unet-like pure transformer for medical image segmentation[C ] //European Conference on Computer Vision. Cham: Springer Nature Switzerland, 2022: 205-218.
FENG G, PAN T F, WU W Y. An online learning behavior analysis based on bayesian network model[J]. Journal of Guangdong University of Technology, 2022, 39(3): 41-48.