Fundus Retinal Vessels Image Segmentation Method Based on Improved U-Net |
| |
Affiliation: | Smart City College, Beijing Union University, Beijing 100101, China |
| |
Abstract: | ObjectivesAlthough the segmentation of retinal vessels in the fundus is of great significance for screening and diagnosing retinal vascular diseases, it remains difficult to detect the low contrast and the information around the lesions provided by retinal vessels in the fundus and to locate and segment micro-vessels in the fine-grained area. To overcome this problem, we propose herein an improved U-Net segmentation method NoL-UNet.Material and methodsThis work introduces NoL-UNet. First of all, the ordinary convolution block of the U-Net network is changed to random dropout convolution blocks, which can better extract the relevant features of the image and effectively alleviate the network overfitting. Next, a NoL-Block attention mechanism added to the bottom of the encoding-decoding structure expands the receptive field and enhances the correlation of pixel information without increasing the number of parameters.ResultsThe proposed method is verified by applying it to the fundus image datasets DRIVE, CHASE_DB1, and HRF. The AUC for DRIVE, CHASE_DB1 and HRF is 0.9861, 0.9891 and 0.9893, Se for DRIVE, CHASE_DB1 and HRF is 0.8489, 0.8809 and 0.8476, and the Acc for DRIVE, CHASE_DB1 and HRF is 0.9697, 0.9826 and 0.9732, respectively. The total number of parameters is 1.70M, and for DRIVE, it takes 0.050s to segment an image.ConclusionOur method is statistically significantly different from the U-Net method, and the improved method shows superior performance with better accuracy and robustness of the model, which has good practical application in auxiliary diagnosis. |
| |
Keywords: | U-Net Fundus retinal blood vessels Non-local Attention mechanism Medical image segmentation |
本文献已被 ScienceDirect 等数据库收录! |
|