Summary of the invention
Of the existing technology in order to solve the problems, such as, at least one embodiment of the present invention provides a kind of for eyeground figure
As the neural network model of blood vessel segmentation.
For this purpose, the embodiment of the present invention proposes a kind of neural network model for eye fundus image blood vessel segmentation, comprising:
Multiple blood vessel feature extraction layers;
The multiple blood vessel feature extraction layer extracts the blood vessel feature in eye fundus image based on default hierarchic sequence;Lowermost layer
Blood vessel feature extraction layer extract eye fundus image in blood vessel feature, obtain characteristic image;The blood vessel feature extraction of not least layer
Layer carries out blood vessel feature extraction based on the characteristic image that low one layer of blood vessel feature extraction layer obtains, and obtains characteristic image;
Multiple blood vessel characteristic processing layers;
Multiple blood vessel feature extraction layers of the multiple blood vessel characteristic processing layer and not least layer connect one to one;Each
The blood vessel characteristic processing layer carries out up-sampling treatment to the characteristic image that the blood vessel feature extraction layer of connection obtains, and obtains adopting
Sampled images;
Multiple blood vessel characteristic optimization layers;
The blood vessel feature extraction layer of the blood vessel characteristic optimization layer connection lowermost layer of lowermost layer, multiple blood vessels of not least layer are special
Multiple blood vessel characteristic processing layers of sign optimization layer and not least layer connect one to one;Top blood vessel characteristic processing layer passes through
Backward short connection is connect with the blood vessel characteristic optimization layer of lowermost layer;Each blood vessel characteristic optimization layer by it is preceding to short connection with
High one layer of blood vessel characteristic optimization layer connection;Each blood vessel characteristic optimization layer, for obtaining the blood vessel characteristic processing of connection
The up-sampling image that layer obtains;The blood vessel characteristic optimization layer of lowermost layer, is also used to obtain the blood vessel feature extraction layer of lowermost layer
Characteristic image;Each blood vessel characteristic optimization layer is also used to successively carry out blood vessel feature extraction to each image of acquisition
Each of and nonlinear processing, obtain the corresponding non-linearization image of each image, and will acquire before passing through to short connection
Image is sent to high one layer of blood vessel characteristic optimization layer;
Blood-vessel image fused layer;
The blood-vessel image fused layer is separately connected each blood vessel characteristic optimization layer;The blood-vessel image fused layer base
In each preset fusion weight of blood vessel characteristic optimization layer, each blood vessel characteristic optimization layer of fusion obtains non-linear
Change image, obtains blood-vessel image.
In some embodiments, the multiple blood vessel feature extraction layer, comprising:
Convolutional layer conv1_2, convolutional layer conv2_2, convolutional layer conv3_3 and the volume of convolutional neural networks model VGG16
Lamination conv4_3.
In some embodiments, the multiple blood vessel characteristic processing layer, comprising:
Second up-sampling layer, third up-sampling layer and the 4th up-sampling layer;
Convolutional layer conv2_2 connection the second up-sampling layer;
The convolutional layer conv3_3 connection third up-samples layer;
Convolutional layer conv4_3 connection the 4th up-sampling layer.
In some embodiments, the multiple blood vessel characteristic optimization layer, comprising: the first convolutional layer, the second convolutional layer, third
Convolutional layer and Volume Four lamination;
Convolutional layer conv1_2 connection first convolutional layer;
The second up-sampling layer connects second convolutional layer;
The third up-sampling layer connects the third convolutional layer;
The 4th up-sampling layer connects the Volume Four lamination;
The 4th up-sampling layer is connect to short connection with first convolutional layer after passing through;
First convolutional layer is connect to short connection with second convolutional layer by preceding;
Second convolutional layer is connect to short connection with the third convolutional layer by preceding;
The third convolutional layer is connect to short connection with the Volume Four lamination by preceding.
In some embodiments, multiple blood vessel characteristic optimization layers, further includes: the first non-linear layer, second non-linear
Layer, third non-linear layer and the 4th non-linear layer;
First convolutional layer connects first non-linear layer;
Second convolutional layer connects second non-linear layer;
The third convolutional layer connects the third non-linear layer;
The Volume Four lamination connects the 4th non-linear layer.
In some embodiments, first non-linear layer, second non-linear layer, the third non-linear layer and institute
It states the 4th non-linear layer and is separately connected the blood-vessel image fused layer.
As it can be seen that at least one embodiment of the embodiment of the present invention, by it is rear to short connection high layer information is passed to it is low
Low level information is passed to high level to short connection by preceding, features at different levels is sufficiently merged, so that blood vessel segmentation is more accurate by layer.
Specific embodiment
In order to make the object, technical scheme and advantages of the embodiment of the invention clearer, below in conjunction with the embodiment of the present invention
In attached drawing, technical scheme in the embodiment of the invention is clearly and completely described, it is clear that described embodiment is
A part of the embodiments of the present invention, instead of all the embodiments.Based on the embodiments of the present invention, ordinary skill people
Member's every other embodiment obtained without making creative work, shall fall within the protection scope of the present invention.
It should be noted that, in this document, the relational terms of such as " first " and " second " or the like are used merely to one
A entity or operation with another entity or operate distinguish, without necessarily requiring or implying these entities or operation it
Between there are any actual relationship or orders.
As shown in Figure 1, the present embodiment discloses a kind of neural network model for the identification of eye fundus image blood vessel, it may include:
Eye fundus image input layer, multiple blood vessel feature extraction layers, multiple blood vessel characteristic processing layers, multiple blood vessel characteristic optimization layers, blood vessel
Image co-registration layer and blood-vessel image output layer.
In the present embodiment, eye fundus image input layer, eye fundus image for receiving input.Although being appreciated that in Fig. 1 only
One eye fundus image input layer is shown, but those skilled in the art can setting be more in neural network model according to actual needs
A eye fundus image input layer, to receive multiple eye fundus images.
In some embodiments, eye fundus image input layer can be omitted.
In the present embodiment, multiple blood vessel feature extraction layers extract the blood vessel spy in eye fundus image based on default hierarchic sequence
Sign.In Fig. 1, the level of blood vessel feature extraction layer successively increases from left to right namely the blood vessel feature extraction layer of the leftmost side is most
The blood vessel feature extraction layer of low layer, the blood vessel feature extraction layer of the rightmost side are top blood vessel feature extraction layers.
In the present embodiment, the blood vessel feature extraction layer of lowermost layer extracts the blood vessel feature in eye fundus image, obtains characteristic pattern
Picture.It is special that the blood vessel feature extraction layer of not least layer based on the characteristic image that low one layer of blood vessel feature extraction layer obtains carries out blood vessel
Sign is extracted, and characteristic image is obtained.
In the present embodiment, blood vessel feature extraction layer realizes blood vessel feature extraction using convolution operation.Convolution is image procossing
Common method, give input picture, to obtained after input picture convolution output image.It is defeated for exporting each pixel in image
Enter the weighted average of pixel in a zonule in image.Average weighted weight is defined by a function, which is known as rolling up
Product core.Convolution kernel used in convolution operation can be by training neural network model to obtain in advance, and training method can be continued to use existing
Technology, this embodiment is not repeated.
In the present embodiment, multiple blood vessel feature extraction layers of multiple blood vessel characteristic processing layers and not least layer, which correspond, to be connected
It connects.Each blood vessel characteristic processing layer carries out up-sampling treatment to the characteristic image that the blood vessel feature extraction layer of connection obtains, and obtains
Up-sample image.
In the present embodiment, the up-sampling treatment of blood vessel characteristic processing layer can continue to use the prior art, and this embodiment is not repeated.
In the present embodiment, the blood vessel feature extraction layer of the blood vessel characteristic optimization layer connection lowermost layer of lowermost layer.Not least layer
Multiple blood vessel characteristic processing layers of multiple blood vessel characteristic optimization layers and not least layer connect one to one.Top blood vessel is special
Sign process layer is connect to short connection with the blood vessel characteristic optimization layer of lowermost layer after passing through.Each blood vessel characteristic optimization layer by it is preceding to
Short connection is connect with high one layer of blood vessel characteristic optimization layer.Each blood vessel characteristic optimization layer, for obtaining the blood vessel feature of connection
The up-sampling image that process layer obtains;The blood vessel characteristic optimization layer of lowermost layer, is also used to obtain the blood vessel feature extraction of lowermost layer
The characteristic image of layer;Each blood vessel characteristic optimization layer is also used to successively carry out blood vessel feature extraction to each image of acquisition
Each of and nonlinear processing, obtain the corresponding non-linearization image of each image, and will acquire before passing through to short connection
Image is sent to high one layer of blood vessel characteristic optimization layer.
In the present embodiment, by preceding to short connection, the semantic information of low layer is passed into high level, it is excellent to reduce blood vessel feature
The semantic gap changed between layer solves lowermost layer so that the non-linearization image that blood vessel characteristic optimization layer obtains is more accurate
The big problem of the obtained characteristic image noise of blood vessel feature extraction layer so that the vessel graph that blood-vessel image fused layer obtains
As more accurate.
In the present embodiment, by rear to short connection, high-rise structural information is passed into low layer, solves top blood
The fuzzy problem of the characteristic image that pipe feature extraction layer obtains, so that the blood-vessel image that blood-vessel image fused layer obtains is more
Accurately.
In the present embodiment, 1 × 1 convolution kernel and convolutional neural networks frame (Convolutional can be based on
Architecture for Fast Feature Embedding, CAFFE) provide attended operation realize before to it is short connection and
Backward short connection, details are not described herein for specific implementation code.The disclosed nerve for being used for eye fundus image blood vessel segmentation of the present embodiment
Network model can be described as supervising full convolutional neural networks model (S-DSN) for short connection depth.
In the present embodiment, blood-vessel image fused layer is separately connected each blood vessel characteristic optimization layer.Blood-vessel image fused layer base
In the preset fusion weight of each blood vessel characteristic optimization layer, the non-linearization image that each blood vessel characteristic optimization layer obtains is merged,
Obtain blood-vessel image.
In the present embodiment, by merging weight to each blood vessel characteristic optimization layer is default, so that blood-vessel image fused layer will
The non-linearization image that each blood vessel characteristic optimization layer obtains is merged according to corresponding fusion weight, is obtained more accurate
Blood-vessel image.Fusion weight can be by training neural network model to obtain in advance, and training method can continue to use the prior art, this implementation
Example repeats no more.
In the present embodiment, blood-vessel image output layer, the blood-vessel image obtained for the image co-registration layer that runs off vascular.
In some embodiments, blood-vessel image output layer can be omitted, and blood-vessel image fused layer can after obtaining blood-vessel image
Directly run off vascular image.
As it can be seen that the disclosed neural network model for eye fundus image blood vessel segmentation of the present embodiment will to short connection after passing through
High layer information passes to low layer, and low level information is passed to high level to short connection by preceding, features at different levels are sufficiently merged, so that blood
Pipe segmentation is more accurate.
Eye fundus image blood vessel is carried out based on the neural network model for being used for eye fundus image blood vessel segmentation disclosed in the present embodiment
The process of segmentation is described as follows step 1 to step 6:
Step 1: eye fundus image input layer receives the eye fundus image of input.
Step 2: multiple blood vessel feature extraction layers extract the blood vessel feature in eye fundus image based on default hierarchic sequence, obtain
Multiple characteristic images.Specifically, the blood vessel feature extraction layer of lowermost layer extracts the blood vessel feature in eye fundus image, obtains characteristic pattern
Picture.It is special that the blood vessel feature extraction layer of not least layer based on the characteristic image that low one layer of blood vessel feature extraction layer obtains carries out blood vessel
Sign is extracted, and characteristic image is obtained.
Step 3: the feature that multiple blood vessel characteristic processing layers respectively obtain multiple blood vessel feature extraction layers of not least layer
Image carries out up-sampling treatment, obtains up-sampling image.
Step 4: the blood vessel feature that the up-sampling image that this layer obtains is passed to connection by each blood vessel characteristic processing layer is excellent
Change layer.The blood vessel characteristic optimization layer of lowermost layer also obtains the characteristic image of the blood vessel feature extraction layer of lowermost layer.Top blood
The up-sampling image that this layer obtains is passed to the blood vessel characteristic optimization layer of lowermost layer by pipe characteristic processing layer to short connection after passing through.
Each blood vessel characteristic optimization layer successively carries out blood vessel feature extraction and nonlinear processing to each image of acquisition, obtains
The corresponding non-linearization image of each image.Each blood vessel characteristic optimization layer connects each image that will acquire hair to short by preceding
Give high one layer of blood vessel characteristic optimization layer.
Step 5: blood-vessel image fused layer is based on the preset fusion weight of each blood vessel characteristic optimization layer, merges each blood vessel
The non-linearization image that characteristic optimization layer obtains, obtains blood-vessel image.
Step 6: blood-vessel image output layer runs off vascular the blood-vessel image that image co-registration layer obtains.
In conclusion the disclosed neural network model for eye fundus image blood vessel segmentation of the present embodiment pass through after to short company
It connects and high layer information is passed into low layer, low level information is passed into high level to short connection by preceding, features at different levels is sufficiently merged, makes
It is more accurate to obtain blood vessel segmentation.The low disadvantage of traditional unsupervised scheme low efficiency, precision is overcome, while having overcome supervision side
In case, neural network model needs successively to extract characteristics of image, loses many useful information, neural network model is caused to learn
Parameter the shortcomings that cannot portraying blood vessel feature completely.
In some embodiments, as shown in Fig. 2, multiple blood vessel feature extraction layers, comprising: convolutional neural networks model
Convolutional layer conv1_2, convolutional layer conv2_2, convolutional layer conv3_3 and the convolutional layer conv4_3 of VGG16.
It should be noted that convolutional layer conv1_2, convolutional layer conv2_2, convolutional layer conv3_3 and convolutional layer in Fig. 2
Between conv4_3 and indirect connection, it is omitted and convolutional layer conv1_2, convolutional layer in neural network model shown in Fig. 2
Conv2_2, convolutional layer conv3_3 and the associated layer of convolutional layer conv4_3.Associated layer is layer used in VGG16, this
The neural network model of embodiment can be continued to use directly, and for the sake of clarity, these associated layers are omitted in Fig. 2.Specifically,
Associated layer specifically: convolutional layer conv1_1, non-linear layer relu1_1, non-linear layer relu1_2, pond layer pool1, volume
Lamination conv2_1, non-linear layer relu2_2, pond layer pool2, convolutional layer conv3_1, non-linear layer relu3_1, convolutional layer
Conv3_2, non-linear layer relu3_2, non-linear layer relu3_3, pond layer pool3, convolutional layer conv4_1, non-linear layer
Relu4_1, convolutional layer conv4_2 and non-linear layer relu4_2.The VGG16 of function and company in to(for) the above associated layer
The relationship of connecing explicitly defines, and the present embodiment is directly continued to use, therefore repeats no more.
In some embodiments, eye fundus image input layer can be omitted, and eye fundus image, which directly inputs, gives convolutional layer conv1_
1。
In the present embodiment, convolutional layer conv1_1, convolutional layer conv2_2, convolutional layer conv3_3 and convolutional layer conv4_3 are again
It can be described as side output (side-output) layer.
Based on upper one embodiment, as shown in Fig. 2, multiple blood vessel characteristic processing layers, comprising: the second up-sampling layer, third
Up-sample layer and the 4th up-sampling layer.
As shown in Fig. 2, convolutional layer conv2_2 connection second up-samples layer;Convolutional layer conv3_3 connection third up-sampling
Layer;Convolutional layer conv4_3 connection the 4th up-samples layer.
Based on upper one embodiment, as shown in Fig. 2, multiple blood vessel characteristic optimization layers, comprising: the first convolutional layer, volume Two
Lamination, third convolutional layer and Volume Four lamination.
Convolutional layer conv1_2 the first convolutional layer of connection.
Second up-sampling layer connects the second convolutional layer.
Third up-samples layer and connects third convolutional layer.
4th up-sampling layer connects Volume Four lamination.
4th up-sampling layer is connect to short connection with the first convolutional layer after passing through.
First convolutional layer is connect to short connection with the second convolutional layer by preceding.
Second convolutional layer is connect to short connection with third convolutional layer by preceding.
Third convolutional layer is connect to short connection with Volume Four lamination by preceding.
Based on upper one embodiment, as shown in Fig. 2, multiple blood vessel characteristic optimization layers, further includes: the first non-linear layer,
Two non-linear layers, third non-linear layer and the 4th non-linear layer;
First convolutional layer connects the first non-linear layer.
Second convolutional layer connects the second non-linear layer.
Third convolutional layer connects third non-linear layer.
Volume Four lamination connects the 4th non-linear layer.
In the present embodiment, the neural network model for eye fundus image blood vessel segmentation is total to there are four blood vessel characteristic optimization layer,
Each blood vessel characteristic optimization layer includes convolutional layer and non-linear layer.For example, the first convolutional layer and the first non-linear layer constitute one
A blood vessel characteristic optimization layer.
Based on upper one embodiment, as shown in Fig. 2, the first non-linear layer, the second non-linear layer, third non-linear layer and
Four non-linear layers are separately connected blood-vessel image fused layer.
Eye fundus image blood vessel segmentation is carried out based on the neural network model for being used for eye fundus image blood vessel segmentation disclosed in Fig. 2
Process is described as follows step 1 to step 6:
Step 1: eye fundus image input layer receives the eye fundus image of input.
Step 2: convolutional layer conv1_2 extracts the blood vessel feature in eye fundus image, obtains characteristic image.Convolutional layer conv2_
2 carry out blood vessel feature extraction based on the obtained characteristic image of convolutional layer conv1_2 again, obtain new characteristic image.Convolutional layer
Conv3_3 carries out blood vessel feature extraction based on the obtained characteristic image of convolutional layer conv2_2 again, obtains new characteristic image.
Convolutional layer conv4_3 carries out blood vessel feature extraction based on the obtained characteristic image of convolutional layer conv3_3 again, obtains new feature
Image.
Step 3: the second up-sampling layer, third up-sampling layer and the 4th up-sampling layer are respectively to convolutional layer conv2_2, convolution
The characteristic image that layer conv3_3 and convolutional layer conv4_3 is obtained carries out up-sampling treatment, obtains up-sampling image.
Step 4: the up-sampling figure that the second up-sampling layer, third up-sampling layer and the 4th up-sampling layer respectively obtain this layer
Image relaying gives the second convolutional layer, third convolutional layer and Volume Four lamination.Convolutional layer conv2_2 passes to obtained characteristic image
First convolutional layer.The up-sampling image that this layer obtains to short connection is passed to the first convolutional layer after passing through by the 4th up-sampling layer.
First convolutional layer, the second convolutional layer, third convolutional layer and Volume Four lamination carry out blood vessel feature to each image of acquisition respectively
It extracts.First convolutional layer, the second convolutional layer and third convolutional layer transmit before passing through respectively to the short each image that will acquire that connects
To high one layer of blood vessel characteristic optimization layer.First non-linear layer, the second non-linear layer, third non-linear layer and the 4th non-linear layer
Each image after the first convolutional layer, the second convolutional layer, third convolutional layer and Volume Four lamination process of convolution is carried out respectively non-
Linearization process obtains non-linearization image.
Step 5: blood-vessel image fused layer is based on the preset fusion weight of each blood vessel characteristic optimization layer, and fusion first is non-thread
The non-linearization image that property layer, the second non-linear layer, third non-linear layer and the 4th non-linear layer obtain, obtains blood-vessel image.
Step 6: blood-vessel image output layer runs off vascular the blood-vessel image that image co-registration layer obtains.
In the present embodiment, using publicly available retinal images blood vessel segmentation data set DRIVE to neural network model
It is trained.DRIVE is made of 40 colored fundus photographs.40 colored fundus photographs are divided into a training set and a test
Collection respectively includes 20 images.
In the present embodiment, the image in training set is pre-processed and obtains that example is trained to shine, for inputting nerve net
Network model.Pretreatment includes: 90 ° of rotations, 180 ° of rotations and 270 ° of rotations, flip horizontal and flip vertical, and to image ruler
One of very little 0.5 times and 0.8 times of scaling is a variety of.
In the present embodiment, the parameter setting during training neural network model is as follows:
For the S-DSN of image level, input data is eye fundus image, and output data is the corresponding blood of eye fundus image
Pipe image.For the S-DSN of patch rank, eye fundus image is divided into 9 patch.9 patch are made after 2 times of up-samplings
The S-DSN of patch rank is inputted for input data.
In the present embodiment, learning rate is set as 10-8, weight attenuation coefficient is 0.0005, momentum 0.9.Side output
(side-output) layer is four layers.Neural network model training learning rate after 25000 iteration reduces by 10 times, 50000 times
Preset parameter is as neural network model final argument after iteration.
It should be noted that, in this document, the terms "include", "comprise" or its any other variant are intended to non-row
His property includes, so that the process, method, article or the device that include a series of elements not only include those elements, and
And further include other elements that are not explicitly listed, or further include for this process, method, article or device institute it is intrinsic
Element.In the absence of more restrictions, the element limited by sentence "including a ...", it is not excluded that including being somebody's turn to do
There is also other identical elements in the process, method of element, article or device.
It will be appreciated by those of skill in the art that although some embodiments described herein include other embodiments is wrapped
Certain features for including rather than other feature, but the combination of the feature of different embodiments mean in the scope of the present invention it
It is interior and form different embodiments.
Although the embodiments of the invention are described in conjunction with the attached drawings, but those skilled in the art can not depart from this hair
Various modifications and variations are made in the case where bright spirit and scope, such modifications and variations are each fallen within by appended claims
Within limited range.