Journal of Jianghan University (Natural Science Edition) ›› 2022, Vol. 50 ›› Issue (4): 87-96.doi: 10.16389/j.cnki.cn42-1737/n.2022.04.011
Previous Articles
CHENG Weidonga,YE Xi*a,WANG Fangb,PING Jingjingc,QIAN Tonghuia,ZHANG Zhiweia
Published:
Contact:
Abstract: A network model combining self-attentive mechanism, superposition loss function,and U-Net network was proposed to achieve accurate segmentation of neuronal images for the current defects of feature ambiguity,high complexity,and lossy edges of electron microscopy neuronal image segmentation. Firstly,the geometric transformation was performed with the original image to enlarge the data set and reduce overfitting. Secondly,the improved self-attention mechanism was used to focus on learning image details to improve the accuracy of model segmentation. Finally,the network performance was improved by appropriately combining Dice Loss with relative entropy (KL scatter). The network experimented on the ISBI 2012 dataset, and its correctness, F1 index,accuracy,and recall reached 0. 930 43,0. 956 79,0. 953 26,and 0. 960 34,respectively. The image segmentation effect was relatively more accurate in overall and detail segmentation,and the cell membrane segmentation was basically unbroken.
Key words: U-Net network, neuron segmentation, attention mechanism, KL scatter, BN layer
CLC Number:
TP391
CHENG Weidong,YE Xi,WANG Fang,PING Jingjing,QIAN Tonghui,ZHANG Zhiwei. Neuron Segmentation Algorithm Based on Improved U-Net Network[J]. Journal of Jianghan University (Natural Science Edition), 2022, 50(4): 87-96.
0 / / Recommend
Add to citation manager EndNote|Reference Manager|ProCite|BibTeX|RefWorks
URL: https://qks.jhun.edu.cn/jhdx_zk/EN/10.16389/j.cnki.cn42-1737/n.2022.04.011
https://qks.jhun.edu.cn/jhdx_zk/EN/Y2022/V50/I4/87