An unsupervised endoscopic image enhancement method

Through the unsupervised learning method, the decomposition network and denoising network model are used, combined with histogram equalization and light smoothing loss, the problems of uneven light and serious noise in endoscopic image enhancement are solved, and efficient image enhancement effect is achieved.

CN119090748BActive Publication Date: 2025-02-07EAST CHINA DIGITAL MEDICAL ENG RES INST
View PDF 2 Cites 0 Cited by

Patent Information

Application Number
CN202411535691.5
Authority / Receiving Office
CN · China
Patent Type
Patents(China)
Current Assignee / Owner
Filing Date
2024-10-31
Publication Date
2025-02-07
Estimated Expiration
2044-10-31

AI Technical Summary

Technical Problem

The existing endoscopic image enhancement methods have problems such as uneven light, low contrast and serious noise. Traditional methods require complex parameter adjustments, and deep learning methods require a large amount of paired training data, which lacks robustness and versatility.

Method used

An unsupervised endoscopic image enhancement method is proposed. By constructing a decomposed network model and a denoising network model, it uses histogram equalization priors, light smoothing loss and reconstruction loss for training, and combines wavelet transformation and attention map to decompose it Reflection map and illumination map, and noise is removed through noise synthesis module and generation adversarial network.

Benefits of technology

It significantly improves the contrast and color reduction of the endoscopic image, removes noise, improves the sharpness and nature of the image, and has high robustness and versatility.

✦ Generated by Eureka AI based on patent content.

Smart Images

  • Figure CN119090748B_ABST
    Figure CN119090748B_ABST
Patent Text Reader

Abstract

The present invention discloses an unsupervised endoscopic image enhancement method. According to the guidance of histogram equalization prior, the method uses a decomposition network to decompose a low-quality endoscopic image into a reflection map and an illumination map, and then inputs the reflection map into a noise decoupling module to obtain a clean reflection map and reflection map noise. Then, two pairs of data, the clean reflection map and the reflection map noise, and the noise-free natural image and the reflection map noise, are input into a noise synthesis module to obtain a synthesized reflection map and a synthesized natural image. Finally, the synthesized natural image is input into a noise decoupling module to obtain a pseudo-natural image. The noise decoupling module is continuously optimized by introducing a discriminator and adversarial loss to obtain the final enhanced image. This method can greatly enhance the contrast and clarity between the input image tissue and blood vessels, and effectively remove a large amount of salt and pepper and Gaussian noise. It provides strong support for subsequent medical diagnosis.
Need to check novelty before this filing date? Find Prior Art

Description

Technical Field

[0001] The invention relates to the technical field of image enhancement, and in particular to an unsupervised endoscopic image enhancement method. Background Art

[0002] In the actual use of endoscopes, due to the complexity of the internal structure of the human body, the concentration of light sources and the limitations of imaging technology, medical endoscope images generally have serious degradation problems such as uneven illumination, low contrast and a lot of noise. These defects not only make it difficult to observe tiny blood vessels and early, inconspicuous lesions, but also seriously affect doctors' diagnosis and treatment, and also reduce the accuracy of some auxiliary diagnostic equipment. Therefore, how to enhance endoscopic images is a problem with great practical application value and research significance.

[0003] Some existing traditional methods include: histogram equalization (HE), gamma correction, unsharp masking technology and Retinex-based methods. Although they can improve some brightness and contrast, they require a series of complex operations to adjust parameters for different scenes and tasks, and the enhancement effect is limited.

[0004] In the field of deep learning, many image quality enhancement methods based on supervised learning of paired images have been proposed. These methods no longer use complex parameter adjustment operations, but usually require a large amount of paired training data. Endoscopic images involve patient privacy, which is difficult to obtain in actual scenarios. In addition, the network model trained using the paired dataset is only effective for endoscopic image enhancement in a certain specific environment, so it lacks robustness and versatility. Summary of the invention

[0005] In view of the deficiencies in the prior art, the present invention provides an unsupervised endoscopic image enhancement method, which aims to solve the problems in the background technology.

[0006] To achieve the above object, the present invention provides the following technical solution: an unsupervised endoscopic image enhancement method, comprising the following steps:

[0007] Step S1: Collect low-quality endoscopic images and high-quality noise-free natural images to construct a dataset;

[0008] Step S2: construct a decomposition network model, input the low-quality endoscopic image into the decomposition network model, and obtain a reflectance map and an illumination map;

[0009] Step S3: performing histogram equalization HE on the low-quality endoscopic image to obtain a histogram equalization prior, and inputting the histogram equalization prior and the reflection map into a pre-trained ConvNeXt network model to obtain intermediate features;

[0010] Step S4: Calculate the loss of the histogram equalization prior according to the intermediate features, and introduce the illumination smoothing loss and reconstruction loss as the total loss of the decomposition network model for training;

[0011] Step S5: constructing a denoising network model, wherein the denoising network model consists of a noise decoupling module and a noise synthesis module;

[0012] Step S6: input the reflection image into the noise decoupling module to obtain a clean reflection image and reflection image noise, input the clean reflection image and reflection image noise as a pair and the noise-free natural image and reflection image noise as a pair into the noise synthesis module respectively to obtain a synthesized reflection image and a synthesized natural image, input the synthesized natural image into the noise decoupling module to obtain a pseudo natural image; introduce content consistency loss and adversarial loss as the total loss of the denoising network model for training, input the pseudo natural image into the trained denoising network model to obtain the final clean reflection image;

[0013] Step S7: construct a discriminator, which uses a generative adversarial network; input the pseudo-natural image and the noise-free natural image obtained in step S6 into the discriminator as a group of image pairs, identify the noise-free natural image and the pseudo-natural image through the discriminator, and introduce the least squares loss as the adversarial loss to optimize the parameters of the noise decoupling module.

[0014] Furthermore, the decomposition network model consists of two convolutional layers and five convolutional blocks; first, the low-quality endoscopic image is normalized, and then the normalized image is passed through the first convolutional layer 5×5, the output of the first convolutional layer 5×5 is subjected to the first wavelet transform to obtain the first high-frequency component and the first low-frequency component respectively, the first low-frequency component is subjected to the second wavelet transform after passing through the first convolutional block to obtain the second high-frequency component and the second low-frequency component, the second low-frequency component is subjected to the third wavelet transform after passing through the second convolutional block to obtain the third high-frequency component and the third low-frequency component, the third high-frequency component is element-wise multiplied with the attention map, and the third The output of the third convolution block of the three low-frequency components and the result of the third high-frequency component and the attention map are concatenated on the channel for the first inverse wavelet transform. The output of the first inverse wavelet transform is concatenated on the channel with the result of the second high-frequency component and the attention map after the fourth convolution block. The result is concatenated on the channel for the second inverse wavelet transform. The output of the second inverse wavelet transform is concatenated on the channel with the result of the first high-frequency component and the attention map after the fifth convolution block. The result is concatenated on the channel for the third inverse wavelet transform. The output of the third inverse wavelet transform is passed through the second convolution layer 5×5 to obtain the reflection map and the illumination map.

[0015] The attention map is composed of the value 1 minus the normalized low-quality endoscopic image.

[0016] Furthermore, the noise decoupling module is composed of multiple convolution blocks; the reflection image first passes through the sixth convolution block, the output of the sixth convolution block is downsampled for the first time, the output of the first downsampling passes through the seventh convolution block, the output of the seventh convolution block is downsampled for the second time, the output of the second downsampling passes through the eighth convolution block, the output of the eighth convolution block is downsampled for the third time, the output of the third downsampling passes through the ninth convolution block, the output of the ninth convolution block is upsampled for the first time, and the output of the first upsampling and the output of the eighth convolution block are subjected to a self-attention mechanism on the channel. The rows are spliced ​​as the input of the tenth convolution block, the output of the tenth convolution block is upsampled for the second time, the output of the second upsampling and the output of the seventh convolution block are spliced ​​on the channel after a self-attention mechanism as the input of the eleventh convolution block, the output of the eleventh convolution block is upsampled for the third time, the output of the third upsampling and the output of the sixth convolution block are spliced ​​on the channel after a self-attention mechanism as the input of the twelfth convolution block, and finally the input of the twelfth convolution block and the reflection map are element-wise added to obtain the denoised image, i.e., the clean reflection map and the reflection map noise.

[0017] Furthermore, each convolution block consists of two 3×3 convolutions, and each convolution is followed by a ReLU activation function.

[0018] Furthermore, the noise synthesis module is composed of six convolutional layers and three residual blocks. First, the clean reflection map or noise-free natural image and the reflection map noise are spliced ​​on the channel as the input of the third convolutional layer 7×7, the output of the third convolutional layer passes through the fourth convolutional layer 3×3 and the fifth convolutional layer 3×3 in sequence, the output of the fifth convolutional layer passes through the first residual block, the output of the first residual block and the output of the fifth convolutional layer are element-wise added as the input of the second residual block, the output of the second residual block and the output of the first residual block are element-wise added as the input of the third residual block, the output of the third residual block passes through the sixth convolutional layer 3×3, the seventh convolutional layer 3×3 and the eighth convolutional layer 7×7 in sequence, and the output of the eighth convolutional layer is element-wise added to the clean reflection map or noise-free natural image to obtain the final synthesized noise image, i.e., the pseudo natural image.

[0019] Furthermore, the output ends of the third convolutional layer 7×7, the fourth convolutional layer 3×3, the fifth convolutional layer 3×3, the sixth convolutional layer 3×3, and the seventh convolutional layer 3×3 are all connected to a LeakyReLU activation function; each residual block is composed of two convolutional layers 3×3 connected in sequence, and the output end of each convolutional layer 3×3 is connected to a LeakyReLU activation function.

[0020] Furthermore, the histogram equalization prior loss , expressed as:

[0021] ;

[0022] In the formula, Indicates intermediate features; Represents a reflection graph; represents the histogram equalization prior; Indicates the calculation of the mean square error between the two;

[0023] The histogram equalization prior loss, illumination smoothing loss and reconstruction loss are introduced as the total loss of the decomposition network model. The weight coefficient of the decomposition network model is set to 0.1, and the hyperparameters Set to 0.01, train the decomposition network model in an unsupervised learning manner, and use the Adam optimizer for optimization; the total loss of the decomposition network model It is expressed as:

[0024] ;

[0025] In the formula, and are the weight coefficients of the decomposition network; represents the reconstruction loss; Represents the lighting smoothing loss.

[0026] Furthermore, we introduce content consistency loss The noise synthesis module is optimized, and the least square loss is introduced as the adversarial loss to optimize the parameters of the noise decoupling module; the denoising network model is jointly trained in an unsupervised learning manner using an unpaired dataset, and the total loss function of the denoising network model is expressed as:

[0027] ;

[0028] In the formula, represents the least squares loss;

[0029] The unpaired data set is formed by combining a public natural image data set and a public endoscopic image data set.

[0030] Furthermore, the specific process of step S1 is: first, a large number of different types of low-quality endoscopic images are collected from the existing publicly available endoscopic datasets, and then a large number of high-quality noise-free natural images are collected from the existing natural image datasets, and the low-quality endoscopic images and high-quality noise-free natural images constitute a dataset.

[0031] Compared with the existing technology, the present invention has the following beneficial effects:

[0032] (1) The present invention introduces a histogram equalization prior to replace the normal endoscopic image as the true value to guide the training of the decomposition network model, so that the decomposed reflectance map and illumination map are more natural than most supervised learning methods and will not produce artifacts.

[0033] (2) In the decomposition network model, the present invention uses wavelet transform and attention map instead of ordinary convolution downsampling operation, so that the detailed texture information of the decomposed endoscopic image is more concentrated in the reflectance map, and the illumination map contains only brightness information as much as possible, avoiding the influence of uneven illumination on the detailed texture information, and can significantly improve the contrast and color reproduction of tissues and blood vessels in the endoscope.

[0034] (3) The present invention transfers the noise in the reflection image to the natural image through the noise synthesis module, and then introduces the generative adversarial loss to effectively remove the salt and pepper noise and Gaussian noise in the reflection image, thereby greatly improving the clarity and naturalness of the blood vessels and tissues in the endoscopic image. BRIEF DESCRIPTION OF THE DRAWINGS

[0035] Figure 1 It is the overall model framework diagram of the algorithm of the present invention.

[0036] Figure 2 This is a structural diagram of the decomposition network model of the algorithm of the present invention.

[0037] Figure 3 This is a structural diagram of the noise decoupling module of the algorithm of the present invention.

[0038] Figure 4 This is a structural diagram of the noise synthesis module of the algorithm of the present invention. DETAILED DESCRIPTION

[0039] like Figure 1 As shown, the present invention provides a technical solution: an unsupervised endoscopic image enhancement method, comprising the following steps:

[0040] Step S1: Collect low-quality endoscopic images and high-quality noise-free natural images to construct a dataset.

[0041] Step S2: construct a decomposition network model, input the low-quality endoscopic image into the decomposition network model, and obtain a reflectance map and an illumination map.

[0042] Step S3: Perform histogram equalization (HE) on the low-quality endoscopic image to obtain a histogram equalization prior, and input the histogram equalization prior and the reflection map into the pre-trained ConvNeXt network model to obtain intermediate features.

[0043] Step S4: Calculate the loss of the histogram equalization prior based on the intermediate features, and introduce the illumination smoothing loss and reconstruction loss as the total loss of the decomposition network model for training.

[0044] Step S5: construct a denoising network model, which is composed of a noise decoupling module and a noise synthesis module.

[0045] Step S6: input the reflection image into the noise decoupling module to obtain a clean reflection image and reflection image noise, input the clean reflection image and reflection image noise as a pair and the noise-free natural image and reflection image noise as a pair into the noise synthesis module respectively to obtain a synthesized reflection image and a synthesized natural image, input the synthesized natural image into the noise decoupling module to obtain a pseudo natural image; introduce content consistency loss and adversarial loss as the total loss of the denoising network model for training, input the pseudo natural image into the trained denoising network model to obtain the final clean reflection image.

[0046] Step S7: Construct a discriminator. The discriminator uses a generative adversarial network (GAN). In order to optimize and judge the denoising ability of the noise decoupling module, the pseudo natural image obtained in step S6 and the input noise-free natural image are input into the discriminator as a set of image pairs. The discriminator is used to identify whether the input is a noise-free natural image or a pseudo natural image. The least squares loss is introduced as the adversarial loss to optimize the parameters of the noise decoupling module, so that the pseudo natural image obtained is closer and closer to the noise-free natural image, thereby deceiving the discriminator. Through the confrontation between the discriminator and the noise decoupling module, the denoising effect of the noise decoupling module is continuously improved.

[0047] The specific process of step S1 is as follows: first, a large number of different types of low-quality endoscopic images are collected from existing publicly available endoscopic datasets, where the low-quality endoscopic images include images of the intestines, abdominal cavity and other parts; then, a large number of high-quality noise-free natural images are collected from existing natural image datasets (LOL and DICM datasets); and the low-quality endoscopic images and high-quality noise-free natural images constitute a dataset.

[0048] Among them, before inputting the low-quality endoscopic images into the decomposition network model, the low-quality endoscopic images are preprocessed, including random cropping to 128*128 pixel size, random flipping and other data enhancement operations.

[0049] like Figure 2As shown, the decomposition network model consists of two convolutional layers and five convolutional blocks; first, the input image (low-quality endoscopic image) is normalized, and then the normalized image is passed through the first convolutional layer (5×5), the output of the first convolutional layer (5×5) is subjected to the first wavelet transform to obtain the first high-frequency component and the first low-frequency component respectively, the first low-frequency component is subjected to the second wavelet transform after passing through the first convolutional block to obtain the second high-frequency component and the second low-frequency component, the second low-frequency component is subjected to the third wavelet transform after passing through the second convolutional block to obtain the third high-frequency component and the third low-frequency component, and the third high-frequency component is element-wise multiplied with the attention map , the output of the third low-frequency component after the third convolution block and the result of the third high-frequency component and the attention map are element-wise multiplied and concatenated on the channel for the first inverse wavelet transform, the output of the first inverse wavelet transform is passed through the fourth convolution block and the result of the second high-frequency component and the attention map are element-wise multiplied and concatenated on the channel for the second inverse wavelet transform, the output of the second inverse wavelet transform is passed through the fifth convolution block and the result of the first high-frequency component and the attention map are element-wise multiplied and concatenated on the channel for the third inverse wavelet transform, and the output of the third inverse wavelet transform is passed through the second convolution layer (5×5) to obtain the reflectance map and the illumination map.

[0050] Among them, the attention map is composed of the value 1 minus the normalized input image (low-quality endoscopic image), and each convolution block is composed of two convolutional layers (3×3), and each convolutional layer is followed by a ReLU activation function.

[0051] Among them, the loss of histogram equalization prior It is expressed as:

[0052] ;

[0053] In the formula, Indicates intermediate features; Represents a reflection graph; represents the histogram equalization prior; It means calculating the mean square error between the two.

[0054] Since image decomposition is an ill-posed decomposition problem, some constraints need to be imposed on the decomposition network model. According to physical prior knowledge, the illumination changes of natural images are continuous and gradual, so the illumination smoothing loss is introduced to constrain the decomposed illumination map to be as smooth as possible in terms of texture details while retaining the structural boundaries. It is expressed as:

[0055] ;

[0056] In the formula, Indicates taking the absolute value; To avoid a small constant with a denominator of 0; It means taking the maximum value between the two. Represents gradient (including horizontal gradient and vertical gradient); represents an illumination diagram; Represents the input low-quality endoscopic image.

[0057] In order to further constrain the decomposition of the decomposition network model so that the decomposed illumination map and reflectance map can reproduce the original input image, the reconstruction loss is introduced. , expressed as:

[0058] ;

[0059] In the formula, Indicates calculation of L1 norm; Represents a reflection graph; Represents element-wise multiplication.

[0060] The histogram equalization prior loss, illumination smoothing loss and reconstruction loss are taken as the total loss of the decomposition network model. The weight coefficient of the decomposition network model is set to 0.1, and the hyperparameters Set to 0.01, train the decomposition network model in an unsupervised learning manner, use the Adam optimizer for optimization, the initial learning rate is 0.0001, and train for 200 epochs; the total loss of the decomposition network model It is expressed as:

[0061] ;

[0062] In the formula, and are the weight coefficients of the decomposition network.

[0063] like Figure 3As shown in FIG. 1 , the noise decoupling module is composed of multiple convolution blocks; the reflection image first passes through the sixth convolution block, the output of the sixth convolution block is downsampled for the first time, the output of the first downsampled is passed through the seventh convolution block, the output of the seventh convolution block is downsampled for the second time, the output of the second downsampled is passed through the eighth convolution block, the output of the eighth convolution block is downsampled for the third time, the output of the third downsampled is passed through the ninth convolution block, the output of the ninth convolution block is upsampled for the first time, and the output of the first upsampled and the output of the eighth convolution block are concatenated on the channel after a self-attention mechanism. The output of the tenth convolution block is then taken as the input of the tenth convolution block, the output of the tenth convolution block is upsampled for the second time, the output of the second upsampling and the output of the seventh convolution block are concatenated on the channel after a self-attention mechanism as the input of the eleventh convolution block, the output of the eleventh convolution block is upsampled for the third time, the output of the third upsampling and the output of the sixth convolution block are concatenated on the channel after a self-attention mechanism as the input of the twelfth convolution block, and finally the input of the twelfth convolution block and the reflection map are element-wise added to obtain the denoised image (clean reflection map and reflection map noise).

[0064] Among them, each convolution block consists of two 3×3 convolutions, and each convolution is followed by a ReLU activation function.

[0065] like Figure 4 As shown in the figure, the noise synthesis module consists of six convolutional layers and three residual blocks. First, the input image (clean reflection map or noise-free natural image) and the input noise (reflection map noise) are concatenated on the channel as the input of the third convolutional layer (7×7). The output of the third convolutional layer passes through the fourth convolutional layer (3×3) and the fifth convolutional layer (3×3) in sequence. The output of the fifth convolutional layer passes through the first residual block. The output of the first residual block is element-wise added to the output of the fifth convolutional layer as the input of the second residual block. The output of the second residual block is element-wise added to the output of the first residual block as the input of the third residual block. The output of the third residual block passes through the sixth convolutional layer (3×3), the seventh convolutional layer (3×3) and the eighth convolutional layer (7×7) in sequence. The output of the eighth convolutional layer is element-wise added to the input image (clean reflection map or noise-free natural image) to obtain the final synthesized noise image (pseudo natural image).

[0066] Among them, the output ends of the third convolutional layer (7×7), the fourth convolutional layer (3×3), the fifth convolutional layer (3×3), the sixth convolutional layer (3×3) and the seventh convolutional layer (3×3) are all connected to a LeakyReLU activation function; each residual block is composed of two convolutional layers (3×3) connected in sequence, and the output end of each convolutional layer (3×3) is connected to a LeakyReLU activation function.

[0067] The noise synthesis module plays the role of noise migration, migrating the noise in the original reflection map to the original natural image. In order to avoid migrating the content and details in the original reflection map to the original natural image, and to make the synthesized reflection map and the synthesized noisy natural image consistent with the original reflection map and the original natural image content structure, the content consistency loss is introduced. Optimize the noise synthesis module, and its calculation formula is as follows:

[0068] ;

[0069] In the formula, Represents a blur kernel size of Gaussian filter operator; Represents SSIM loss, which is used to constrain the structural information of the image; and Represent the original reflection map and the synthetic reflection map respectively; and is the weight coefficient of content consistency loss. Content consistency loss consists of two parts: L1 loss of Gaussian filtering and structural similarity loss. The former constrains the low-frequency information of the image, and the latter constrains the structural information of the image. The parameters of the noise synthesis module are continuously optimized through content consistency loss to generate high-quality synthetic noise images.

[0070] Introducing least squares loss as adversarial loss It is expressed as:

[0071] ;

[0072] In the formula, is a noise-free natural image; It is a pseudo-natural image; is the discriminator; is the generator. The parameters of the noise decoupling module can be continuously optimized through adversarial loss, so that the denoising effect is optimal.

[0073] Among them, the denoising network model uses an unpaired dataset for joint training in an unsupervised learning manner, and uses the total loss function of the denoising network model: , where the weight coefficient of content consistency loss is and are set to 0.1 and 1 respectively, and the kernel size The training was set to 5, 200 epochs were trained, the learning rate was 0.0001, the stochastic gradient descent algorithm (SGD) was used for optimization, and the optimal model parameters were saved after all training was completed.

[0074] The unpaired dataset is formed by combining a public natural image dataset (LOL natural image dataset, DICM natural image dataset) with a public endoscopic image dataset (SUN endoscopic image dataset).

[0075] Although embodiments of the present invention have been shown and described, it will be appreciated by those skilled in the art that various changes, modifications, substitutions and variations may be made to the embodiments without departing from the principles and spirit of the present invention, and that the scope of the present invention is defined by the appended claims and their equivalents.

Claims

1. An unsupervised endoscopic image enhancement method, characterized in that: The steps include: Step S1: Collect low-quality endoscopic images and high-quality noise-free natural images to construct a dataset; Step S2: construct a decomposition network model, input the low-quality endoscopic image into the decomposition network model, and obtain a reflectance map and an illumination map; Step S3: performing histogram equalization HE on the low-quality endoscopic image to obtain a histogram equalization prior, and inputting the histogram equalization prior and the reflection map into a pre-trained ConvNeXt network model to obtain intermediate features; Step S4: Calculate the loss of the histogram equalization prior according to the intermediate features, and introduce the illumination smoothing loss and reconstruction loss as the total loss of the decomposition network model for training; Step S5: constructing a denoising network model, wherein the denoising network model consists of a noise decoupling module and a noise synthesis module; Step S6: inputting the reflection image into a noise decoupling module to obtain a clean reflection image and reflection image noise, inputting the clean reflection image and reflection image noise as a pair and the noise-free natural image and reflection image noise as a pair into a noise synthesis module respectively to obtain a synthesized reflection image and a synthesized natural image, and inputting the synthesized natural image into the noise decoupling module to obtain a pseudo natural image; Build a discriminator, which uses a generative adversarial network; The obtained pseudo natural image and the noise-free natural image are input into the discriminator as a group of image pairs, and the noise-free natural image and the pseudo natural image are identified by the discriminator; Content consistency loss and adversarial loss are introduced as the total loss of the denoising network model. The denoising network model is jointly trained using an unpaired dataset in an unsupervised learning manner. The reflection map is input into the trained denoising network model to obtain the final clean reflection map. The unpaired data set is formed by combining a public natural image data set and a public endoscopic image data set; Introducing content consistency loss Optimize the noise synthesis module, expressed as: ; In the formula, Represents a blur kernel size of Gaussian filter operator; represents SSIM loss; and Represent the original reflection map and the synthetic reflection map respectively; and is the weight coefficient of content consistency loss; Introducing least squares loss as adversarial loss Optimize the parameters of the noise decoupling module, expressed as: ; In the formula, is a noise-free natural image; It is a pseudo-natural image; is the discriminator; is a generator; The total loss function of the denoising network model It is expressed as: 。 2. The unsupervised endoscopic image enhancement method according to claim 1, characterized in that: The decomposition network model consists of two convolutional layers and five convolutional blocks; first, the low-quality endoscopic image is normalized, and then the normalized image is passed through the first convolutional layer 5×5, the output of the first convolutional layer 5×5 is subjected to the first wavelet transform to obtain the first high-frequency component and the first low-frequency component respectively, the first low-frequency component is subjected to the second wavelet transform after passing through the first convolutional block to obtain the second high-frequency component and the second low-frequency component, the second low-frequency component is subjected to the third wavelet transform after passing through the second convolutional block to obtain the third high-frequency component and the third low-frequency component, the third high-frequency component is element-wise multiplied with the attention map, and the third low-frequency component is converted into the third high-frequency component. The output of the third convolution block and the result of the third element-wise multiplication of the high-frequency component and the attention map are concatenated on the channel for the first inverse wavelet transform. The output of the first inverse wavelet transform is concatenated on the channel with the result obtained by passing the fourth convolution block and the result obtained by element-wise multiplication of the second high-frequency component and the attention map for the second inverse wavelet transform. The output of the second inverse wavelet transform is concatenated on the channel with the result obtained by passing the fifth convolution block and the result obtained by element-wise multiplication of the first high-frequency component and the attention map for the third inverse wavelet transform. The output of the third inverse wavelet transform is passed through the second convolution layer 5×5 to obtain a reflectance map and an illumination map. The attention map is composed of the value 1 minus the normalized low-quality endoscopic image.

3. The unsupervised endoscopic image enhancement method according to claim 2, characterized in that: The noise decoupling module is composed of multiple convolution blocks; including the sixth convolution block, the output end of the sixth convolution block is respectively connected to the self-attention mechanism and the first downsampling, the output end of the first downsampling is connected to the seventh convolution block, the output end of the seventh convolution block is respectively connected to the self-attention mechanism and the second downsampling, the output end of the second downsampling is connected to the eighth convolution block, the output end of the eighth convolution block is respectively connected to the self-attention mechanism and the third downsampling, the output end of the third downsampling is connected to the ninth convolution block, the output end of the ninth convolution block is connected to the first upsampling, the output of the first upsampling and the output of the eighth convolution block are spliced ​​through the output of the self-attention mechanism and then input into the tenth convolution block, the output end of the tenth convolution block is connected to the second upsampling, the output of the second upsampling and the output of the seventh convolution block are spliced ​​through the output of the self-attention mechanism and then input into the eleventh convolution block, the output end of the eleventh convolution block is connected to the third upsampling, the output of the third upsampling and the output of the sixth convolution block are spliced ​​through the output of the self-attention mechanism and then input into the twelfth convolution block, the output of the twelfth convolution block is added to the input element of the sixth convolution block to obtain the output of the noise decoupling module.

4. The unsupervised endoscopic image enhancement method according to claim 3, characterized in that: Each convolution block consists of two 3×3 convolutions, and each convolution is followed by a ReLU activation function.

5. The unsupervised endoscopic image enhancement method according to claim 4, characterized in that: The noise synthesis module is composed of six convolutional layers and three residual blocks, including a third convolutional layer 7×7, the input of the third convolutional layer 7×7 is spliced ​​by the input image and the input noise, the output end of the third convolutional layer 7×7 is connected to a LeakyReLU activation function and then connected to the fourth convolutional layer 3×3, the output end of the fourth convolutional layer 3×3 is connected to a LeakyReLU activation function and then connected to the fifth convolutional layer 3×3, the output end of the fifth convolutional layer 3×3 is connected to a LeakyReLU activation function and then connected to the first residual block, and the fifth convolutional layer 3×3 is connected to a LeakyReLU activation function. The output after multiplication and the output of the first residual block are element-wise added as the input of the second residual block, and the output of the second residual block and the output of the first residual block are element-wise added as the input of the third residual block. The output end of the third residual block is connected to the sixth convolutional layer 3×3, the output end of the sixth convolutional layer 3×3 is connected to a LeakyReLU activation function and then connected to the seventh convolutional layer 3×3, the output end of the seventh convolutional layer 3×3 is connected to a LeakyReLU activation function and then connected to the eighth convolutional layer 7×7, and the output of the eighth convolutional layer 7×7 is added to the input image elements to obtain the output of the noise synthesis module; Each residual block is composed of two 3×3 convolutional layers connected in sequence, and the output of each 3×3 convolutional layer is connected to a LeakyReLU activation function.

6. The unsupervised endoscopic image enhancement method according to claim 5, characterized in that: The histogram equalization prior loss , expressed as: ; In the formula, Indicates intermediate features; Represents a reflection graph; represents the histogram equalization prior; Indicates the calculation of the mean square error between the two; The histogram equalization prior loss, illumination smoothing loss and reconstruction loss are introduced as the total loss of the decomposition network model. The weight coefficient of the decomposition network model is set to 0.1, and the hyperparameters Set to 0.01, train the decomposition network model in an unsupervised learning manner, and use the Adam optimizer for optimization; Decomposing the total loss of the network model It is expressed as: ; In the formula, and are the weight coefficients of the decomposition network; represents the reconstruction loss; Represents the lighting smoothing loss.

7. The unsupervised endoscopic image enhancement method according to claim 1, characterized in that: The specific process of step S1 is: first, a large number of different types of low-quality endoscopic images are collected from the existing public endoscopic datasets, and then a large number of high-quality noise-free natural images are collected from the existing natural image datasets, and the low-quality endoscopic images and high-quality noise-free natural images constitute a dataset.

Citation Information

Patent Citations

  • Improved Retinex-Net low illumination and scotopic vision image enhancement method

    CN115018717A

  • Self-adaptive low-light noise image enhancement method and system based on zero-order learning

    CN116579944A