中国光学(中英文)2024,Vol.17Issue(4):971-981,11.DOI:10.37188/CO.2023-0186
基于特征图金字塔的冠脉造影图像血管分割方法
Coronary artery angiography image vessel segmentation method based on feature pyramid network
摘要
Abstract
To address issues such as uneven illumination in coronary angiography images,low contrast between vascular structures and background regions,and the complexity of coronary vascular topology,we establish a coronary angiography vascular segmentation annotation dataset.Additionally,we propose a coronary angiography image vascular segmentation model based on the feature map pyramid.On the basis of the U-Net architecture,this model was improved and optimized.First,the first convolutional layer in the U-Net encoding part was replaced with a 7×7 convolutional layer to increase the receptive field of each layer.Modified ConvNeXt blocks were added to the encoding and decoding layers to enhance the network's ability to extract deeper-level features.Second,a Group Attention(GA)mechanism module was designed and incor-porated at the U-Net skip connection to strengthen the features extracted from the encoding part,addressing semantic gaps between the encoder and decoder.Finally,a Pyramid Feature Concatenation(PFC)module was designed at the U-Net decoder,which fused features from different scales.Squeeze-and-Excitaton(SE)attention mechanisms were added to each layer of the PFC to filter out effective information from the feature maps.The loss function of the network is weighted based on the outputs of the PFC module at each layer,serving to supervise the feature extraction process across different layers of the network.The test results of this model on the test set are as follows:the Dice coefficient is 0.8843 and the Jaccard coefficient is 0.7926.Experimental results indicate that this model is highly robust in coronary vascular segmentation,more effect-ively suppressing noise under low contrast and achieving better segmentation results for coronary vessels when compared to other methods.关键词
冠脉造影/血管分割/特征金字塔网络/注意力机制/U-NetKey words
coronary angiography/vessel segmentation/feature pyramid network/attention mechanism/U-Net分类
信息技术与安全科学引用本文复制引用
郭昊虎,高若谦,葛明锋,董文飞,刘炎,赵旭峰..基于特征图金字塔的冠脉造影图像血管分割方法[J].中国光学(中英文),2024,17(4):971-981,11.基金项目
国家重点研发计划(No.2021YFC2500500) (No.2021YFC2500500)
吉林省与中国科学院科技合作高新技术产业化专项资金项目(No.2023SYHZ0037)Supported by the National Key R&D Program of China(No.2021YFC2500500) (No.2023SYHZ0037)
Science and Technology Co-operation Special Project,Jilin Province and Chinese Academy of Sciences(No.2023SYHZ0037) (No.2023SYHZ0037)