Application of an attention U-Net incorporating transfer learning for optic disc and cup segmentation.

Citation metadata

From: Signal, Image and Video Processing(Vol. 15, Issue 5)
Publisher: Springer
Document Type: Report; Brief article
Length: 292 words

Document controls

Main content

Abstract :

Keywords: Image segmentation; Optic disc and cup; U-Net; Attention; Transfer learning Abstract Optic disc (OD) and optic cup (OC) regional parameters are of utmost importance in the early diagnosis of glaucoma. Improving the accuracy of OD/OC segmentation results and parameter extraction in colour fundus images plays a very important role in early glaucoma screening. To improve the accuracy and inference speed of fundus image segmentation, an algorithm for fundus image segmentation based on an attention U-Net with transfer learning is proposed in this paper. First, an attention gate was added between the encoder and decoder of U-Net to focus on the target areas, thus forming the architecture of the attention U-Net. Then, after the network had been trained on the DRIONS-DB dataset to partially obtain the weights of the encoder, it was trained on the Drishti-GS dataset to further modify the weights. Finally, the trained attention U-Net model incorporating transfer learning was used to segment fundus images. OD/OC extraction using this method shows obvious advantages in model parameter quantity and inference time compared with existing algorithms, the parameter quantity is much smaller than that of existing algorithms, and the model inference time is 0.33 s, representing a reduction of more than 50%. The proposed method can be applied to a fundus image dataset with only a small number of labels. Whilst offering fast OD/OC segmentation, it also guarantees a relatively high segmentation accuracy. Author Affiliation: (1) School of Electrical and Information Engineering, North Minzu University, 750011, Yinchuan, Ningxia, China (2) Basic Experimental Teaching and Engineering Training Center, North Minzu University, 750011, Yinchuan, Ningxia, China (3) School of Information Engineering, Ningxia University, 750011, Yinchuan, Ningxia, China (d) wei_hc@nun.edu.cn Article History: Registration Date: 11/05/2020 Received Date: 04/01/2020 Accepted Date: 11/04/2020 Online Date: 11/22/2020 Byline:

Source Citation

Source Citation   

Gale Document Number: GALE|A666667542