Computer Science ›› 2024, Vol. 51 ›› Issue (11A): 231000071-9.doi: 10.11896/jsjkx.231000071

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Study on Deep Learning Algorithm for Foreground Subject Segmentation of Non-specific CategoryImages

CHEN Xianglong, LI Haijun   

  1. School of Information and Intelligent Engineering,University of Sanya,Sanya,Hainan 572022,China
    Academician Guoliang Chen Team Innovation Center,University of Sanya,Sanya,Hainan 572022,China
  • Online:2024-11-16 Published:2024-11-13
  • About author:CHEN Xianglong,born in 2001,postgraduate,is a member of CCF(No.Q5078G).His main research interests include computer vision and data mining.
    LI Haijun,born in 1968,Ph.D,associate professor,master's supervisor,is a member of CCF(No.F7747M).His main research interests include compu-ter vision and data mining.

Abstract: By incorporating SENet channel attention mechanism on the basis of Mobile Unet network,the image foreground subject se-gmentation algorithm is improved.The algorithm introduces deep separable convolution to reduce the number of model parameters,while utilizing skip connections and multi-scale feature fusion to improve the segmentation accuracy of the model.Du-ring the training process,a spatial pyramid pooling module with hollow convolution is used to increase the receptive field and improve the model's recognition ability for large-scale objects.Experimental results show that the improved algorithm achieves 96% MIOU(Modular Input/Output Un-it) segmentation accuracy on the PASCAL VOC2012 dataset,with an accuracy rate of 0.971,which is superior to various existing image segmentation algorithms,such as the FCN fully convolutional neural network algorithm.In terms of speed,the processing time of the model for each image is between 1.7 s and 2.5 s.The improved algorithm has a faster inference speed compared to traditional fully convolutional neural networks,making it suitable for real-time image segmentation on mobile devices.Through comparative experiments,the effectiveness of the Mobile Unet models before and after the improvement,as well as the FCN model,in foreground subject segmentation of images under bright and dim conditions is compared,and the conclusion is drawn that the improved Mobile Unet model has the best performance.Finally,the algorithm is deployed,a GUI visualization operation interface is designed,and an.exe executable file is generated.

Key words: Subject segmentation, Neural network, Receptive field, Parameter quantity, Segmentation accuracy

CLC Number: 

  • TP183
[1]LIU Y,YANG S.Shadow segmentation of indoor moving objects based on improved UNet network [J].Computer System Application,2022,31(12):412-419.
[2]MENG M Z,LI L,HE G Y,et al.Preliminary study on the deg-radation effect of MobileNetV2 on four types of breast X-ray BI-RADS lesions [J].Journal of Clinical Radiology,2022,41(10):1868-1873.
[3]GUOFENG M.A Small Target Detection Method Based on the Improved FCN Model[J].Advances in Multimedia,2022,2022.
[4]YIMING L,ZHENGLE W,RUJIA W,et al.Flooding-basedMobileNet toidentify cucumber diseases from leaf images in natural scenes[J].Computers and Electronics in Agriculture,2023,213.
[5]CHEN L C,PAPANDREOV G,KOKKINOS I,et al.DeepLab:Semantic Image Segmentation with Deep Convolutional Nets,Atrous Convolution,and Fully Connected CRFs.[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2018,40(4):834-838.
[6]WU D,ZHAO J,WANG Z.AM-PSPNet:Pyramid Scene Parsing Network Based on Attentional Mechanism for Image Semantic Segmentation[C]//ICPCSEE Steering Committee.Abstracts of the 8th International Conference of Pioneering Computer Scientists,Engineers and Educators(ICPCSEE 2022).Part I.Springer,2022:444.
[7]ZHANG X,YAO Q A,ZHAO J,et al.Overview of Full Convo-lutional Neural Network Image Semantic Segmentation Methods [J].Computer Engineering and Applications,2022,58(8):45-57.
[8]HUANG P,ZHENG Q,LIANG O.Overview of Image Segmentation Methods [J].Journal of Wuhan University(Science Edition),2020,66(6):519-531.
[9]HUANG S P,LIU H N,ZHOU K S,et al.Zebra crossing segmentation based on improved Unet [J].Intelligent Computer and Applications,2020,10(11):61-64,69.
[10]SUN L X.Coal gangue image recognition based on convolutional neural network [J].Computer Knowledge and Technology,2020,16(21):16-18,22.
[11]LI Y Q.Research and Implementation of Image Saliency Detection Algorithm Based on Attention Mechanism and U-Net [D].Beijing:Beijing Jiaotong University,2020.
[12]WANG P,GAO C,ZHU L,et al.Ischemic stroke lesion segmentation algorithm based on 3D deep residual network and cascaded U-Net [J].Computer Application,2019,39(11):3274-3279.
[13]DING R J,GAO F F,XING L.An Intelligent Routing Strategy for the Internet of Things Based on Deep Reinforcement Learning [J].Journal of the Internet of Things,2019,3(2):56-63.
[14]TIAN X,WANG L,DING Q.Overview of Image Semantic Segmentation Methods Based on Deep Learning [J].Journal of Software Science,2019,30(2):440-468.
[15]ZHANG M Y.Research on Image Segmentation Based on Deep Learning [D].Jilin:Jilin University,2017.
[16]CHEN H X.Image Semantic Segmentation Based on Convolutional Neural Networks [D].Hangzhou:Zhejiang University,2016.
[17]LIU S T,YIN F L.Image segmentation methods based on graph cuts and their new progress [J].Journal of Automation,2012,38(6):911-922.
[18]HE J,GE H,WANG Y F.Overview of Image Segmentation Algorithms [J].Computer Engineering and Science,2009,31(12):58-61.
[1] CHEN Liang, SUN Cong. Deep-learning Based DKOM Attack Detection for Linux System [J]. Computer Science, 2024, 51(9): 383-392.
[2] ZHU Fukun, TENG Zhen, SHAO Wenze, GE Qi, SUN Yubao. Semantic-guided Neural Network Critical Data Routing Path [J]. Computer Science, 2024, 51(9): 155-161.
[3] TANG Ying, WANG Baohui. Study on SSL/TLS Encrypted Malicious Traffic Detection Algorithm Based on Graph Neural Networks [J]. Computer Science, 2024, 51(9): 365-370.
[4] LIU Renyu, CHEN Xin, SHANG Honghui, ZHANG Yunquan. Optimization of Atomic Kinetics Monte Carlo Program TensorKMC Based on Machine Learning Atomic Potential Functions [J]. Computer Science, 2024, 51(9): 23-30.
[5] CHEN Shanshan, YAO Subin. Study on Recommendation Algorithms Based on Knowledge Graph and Neighbor PerceptionAttention Mechanism [J]. Computer Science, 2024, 51(8): 313-323.
[6] CHENG Xuefeng, DONG Minggang. Dynamic Multi-objective Optimization Algorithm Based on RNN Information Accumulation [J]. Computer Science, 2024, 51(8): 333-344.
[7] ZENG Zihui, LI Chaoyang, LIAO Qing. Multivariate Time Series Anomaly Detection Algorithm in Missing Value Scenario [J]. Computer Science, 2024, 51(7): 108-115.
[8] HU Haibo, YANG Dan, NIE Tiezheng, KOU Yue. Graph Contrastive Learning Incorporating Multi-influence and Preference for Social Recommendation [J]. Computer Science, 2024, 51(7): 146-155.
[9] HAN Bing, DENG Lixiang, ZHENG Yi, REN Shuang. Survey of 3D Point Clouds Upsampling Methods [J]. Computer Science, 2024, 51(7): 167-196.
[10] XU Xiaohua, ZHOU Zhangbing, HU Zhongxu, LIN Shixun, YU Zhenjie. Lightweight Deep Neural Network Models for Edge Intelligence:A Survey [J]. Computer Science, 2024, 51(7): 257-271.
[11] WEI Ziang, PENG Jian, HUANG Feihu, JU Shenggen. Text Classification Method Based on Multi Graph Convolution and Hierarchical Pooling [J]. Computer Science, 2024, 51(7): 303-309.
[12] SUN Yang, DING Jianwei, ZHANG Qi, WEI Huiwen, TIAN Bowen. Study on Super-resolution Image Reconstruction Using Residual Feature Aggregation NetworkBased on Attention Mechanism [J]. Computer Science, 2024, 51(6A): 230600039-6.
[13] QUE Yue, GAN Menghan, LIU Zhiwei. Object Detection with Receptive Field Expansion and Multi-branch Aggregation [J]. Computer Science, 2024, 51(6A): 230600151-6.
[14] YUAN Zhen, LIU Jinfeng. Denoising Autoencoders Based on Lossy Compress Coding [J]. Computer Science, 2024, 51(6A): 230400172-7.
[15] DAI Yongdong, JIN Yang, DAI Yufan, FU Jing, WANG Maofei, LIU Xi. Study on Intelligent Defect Recognition Algorithm of Aerial Insulator Image [J]. Computer Science, 2024, 51(6A): 230700172-5.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!