AUTOMATIC DETECTION AND GRADING OF DIABETIC MACULAR EDEMA BASED ON A DEEP NEURAL NETWORK

摘要

Purpose: To solve the problem of automatic grading of macular edema in retinal images in a more stable and reliable way and reduce the workload of ophthalmologists, an automatic detection and grading method of diabetic macular edema based on a deep neural network is proposed. Methods: The enhanced green channels of fundus images are input into the YOLO network for training and testing. Diabetic macular edema is graded according to the distance of the macula and hard exudate. We used multiscale feature fusion to form more comprehensive features on different grain images to improve the effect of hard exudate detection. We adopted K-means++ algorithm to cluster anchor box size and use LGIoU loss of the original network to guide the regression of hard exudate bounding box and improve the regression accuracy of anchor boxes. We increased the diversity of samples for sample training by data augmentation, including cropping, flipping, and rotating of fundus images, so that each batch of training data can better represent the distribution of samples. Results: The detection accuracy of the proposed method can reach 96% on the MESSIDOR data set. The detection rates of hard exudate with high, median, and low probability are 100%, 79.12%, and 60.40%, respectively. Conclusion: The proposed method exhibits a very good detection stability on healthy and diseased fundus images.