The automatic segmentation of main vessels on X-ray angiography (XRA) images is of great importance in the smart coronary artery disease diagnosis system. However, existing methods have been developed to this task, but these methods have difficulty in recognizing the coronary artery structure in XRA images. Main vessel segmentation is still a challenging task due to the diversity and small-size region of the vessel in the XRA images. In this study, we propose a robust method for main vessel segmentation by using deep learning architectures with fully convolutional networks. Four deep learning models based on the UNet architecture are evaluated on a clinical dataset, which consists of 3200 X-ray angiography images collected from 1118 patients. Using the precision (Pre), recall (Re), and F1 score (F1) as evaluation metrics, the average Pre, Re, and F1 for main vessel segmentation in the entire experimental dataset is 0.901, 0.898, and 0.900, respectively. 89.8% of the images exhibited a high F1 score >0.8. For the main vessel segmentation in XRA images, our deep learning methods demonstrated that vessels could be segmented in real time with a more optimized implementation, to further facilitate the online diagnosis in smart medical.
Comparative Study of Deep Learning Models for Automatic Coronary Stenosis Detection in X-ray Angiography