The invention discloses an attention mechanism-based in-depth learning
diabetic retinopathy classification method comprising the following steps: a series of eye ground images are chosen as
original data samples which are then subjected to normalization preprocessing operation, the preprocessed
original data samples are divided into a
training set and a testing set after being
cut, a main neutralnetwork is subjected to parameter initializing and
fine tuning operation, images of the
training set are input into the main
neutral network and then are trained, and a characteristic graph is generated; parameters of the main
neutral network are fixed, the images of the
training set are adopted for training an
attention network,
pathology candidate zone degree graphs are output and normalized, anattention graph is obtained, an attention mechanism is obtained after the attention graph is multiplied by the characteristic graph, an obtained result of the attention mechanism is input into the main
neutral network, the images of the training set are adopted for training operation, and finally a
diabetic retinopathy grade classification model is obtained. Via the method disclosed in the invention, the attention mechanism is introduced, a
diabetic retinopathy zone
data set is used for training the same, and information characteristics of a
retinopathy zone is enhanced while original networkcharacteristics are reserved.