Pipeline disease image classification method based on multi-label convolutional neural network
A technology of convolutional neural network and classification method, which is applied in the field of computer digital image processing and deep learning algorithm based on convolutional neural network, which can solve the problems of different discrimination sensitivity, difficulty in obtaining high-quality images, and low classification accuracy. , to achieve a wide range of applications, rich types, and improved accuracy.
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0046] A pipeline disease image classification method based on a multi-label convolutional neural network. The steps of the pipeline disease image classification method are as follows:
[0047] Step 1: Collect the pipe endoscope detection video, and extract the image frames in the pipe endoscope detection video;
[0048] Step 2: Calculate the timestamp feature of each image;
[0049] Step 3: Send a part of the image frames collected in step 1 to the multi-label convolutional neural network model for training, and obtain a multi-label convolutional neural network model that can correctly classify the types of pipeline diseases;
[0050] Step 4: Use the trained multi-label convolutional neural network model to detect the endoscopic image of the pipeline to be detected, and then the multi-label convolutional neural network model will output the one-hot code, and determine the existing pipeline disease type according to the one-hot code.
[0051] The multi-label convolutional neural network...
Embodiment 2
[0061] This embodiment combines the above-mentioned embodiments to further illustrate the content, so that those skilled in the art can more clearly understand the implementation of the present invention. The present invention adds multi-label classification to the existing Inception-ResNet-v2 network. Layer, realize the classification function of a variety of pipeline disease images, the present invention replaces the SoftMax classifier with a multi-label classification layer, so that Inception-ResNet-v2 has a multi-label classification function, thereby maximizing detection of the expected disease types ,Such as figure 2 Shown.
[0062] Among them, the random inactivation layer of the upper Inception-ResNet-v2 network structure will output a feature vector of 1792 dimensions, corresponding to figure 2 The X layer in. The present invention adds a one-dimensional feature to the back of this vector, such as figure 2 As shown by the light gray box in, this feature is the time...
Embodiment 3
[0069] This embodiment combines the above-mentioned embodiments to further illustrate the content, so that those skilled in the art can more clearly understand the implementation of the present invention. The present invention adds multi-label classification to the existing Inception-ResNet-v2 network. Layer, realize the classification function of a variety of pipeline disease images, the present invention replaces the SoftMax classifier with a multi-label classification layer, so that Inception-ResNet-v2 has a multi-label classification function, thereby maximizing detection of the expected disease types .
[0070] Such as figure 2 As shown, the X layer is the output vector of the random inactivation layer in the upper Inception-ResNet-v2 network structure shown.
[0071] The random inactivation layer of the original Inception-ResNet-v2 network structure only outputs the first 1792 dimensional information, and the present invention adds a dimension of time information on this ba...
PUM
Login to View More Abstract
Description
Claims
Application Information
Login to View More 


