The invention discloses an OCT
retina image classification method based on an improved ResNet-34 network, and relates to the technical field of
retina medical image classification, and the method comprises the steps: obtaining a
retina OCT image
data set, and carrying out the preprocessing of an image; an improved ResNet-34
network model is constructed, a CBAM attention module is embedded in a residual block of the improved ResNet-34
network model, a CBAM-
Block structure is formed, a channel attention module and a space attention module are combined in a series connection mode, channel feature response of a
focus area is self-adaptively enhanced, and a
key space position is focused; dynamically adjusting
floating point calculation precision in a network training process by adopting an automatic mixing precision training technology; pre-training weight initialization network parameters based on an ImageNet
data set are loaded, and the model is finely adjusted through transfer learning; and training a network by using the preprocessed
data set, and outputting a retina
pathological state
classification result. According to the method, through a triple collaborative optimization mechanism of a CBAM-Block residual structure, automatic mixing precision training and transfer learning, the precision and efficiency of retina OCT image classification are remarkably improved.