[go: up one dir, main page]
More Web Proxy on the site http://driver.im/

WO2022178975A1 - Noise field-based image noise reduction method and apparatus, device, and storage medium - Google Patents

Noise field-based image noise reduction method and apparatus, device, and storage medium Download PDF

Info

Publication number
WO2022178975A1
WO2022178975A1 PCT/CN2021/090430 CN2021090430W WO2022178975A1 WO 2022178975 A1 WO2022178975 A1 WO 2022178975A1 CN 2021090430 W CN2021090430 W CN 2021090430W WO 2022178975 A1 WO2022178975 A1 WO 2022178975A1
Authority
WO
WIPO (PCT)
Prior art keywords
noise
image
noise distribution
denoised
model
Prior art date
Application number
PCT/CN2021/090430
Other languages
French (fr)
Chinese (zh)
Inventor
陈昊
Original Assignee
平安科技(深圳)有限公司
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by 平安科技(深圳)有限公司 filed Critical 平安科技(深圳)有限公司
Publication of WO2022178975A1 publication Critical patent/WO2022178975A1/en

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T5/00Image enhancement or restoration
    • G06T5/70Denoising; Smoothing
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • G06N3/084Backpropagation, e.g. using gradient descent
    • YGENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
    • Y02TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
    • Y02TCLIMATE CHANGE MITIGATION TECHNOLOGIES RELATED TO TRANSPORTATION
    • Y02T90/00Enabling technologies or technologies with a potential or indirect contribution to GHG emissions mitigation

Definitions

  • the present application relates to the technical field of artificial intelligence, and in particular, to a noise field-based image noise reduction method, device, device, and storage medium.
  • a necessary step in the process of optimizing image quality is to denoise the image.
  • image denoising is divided into two ideas: one is to use artificially designed models to denoise images; the other is to use deep learning estimation for image noise reduction.
  • the former is often not applicable in most cases because the designed model can only be aimed at a specific situation; the image noise reduction method of deep learning convolutional neural network emerging in recent years, although it has good adaptability,
  • the actual principle of the deep learning prediction method is to make the model predict the image as a whole. After the image is predicted, the overall image is denoised according to the prediction result. However, this overall denoising method is often localized Regions cause distortion of the image.
  • the purpose of the embodiments of the present application is to propose an image noise reduction method, device, computer equipment and storage medium based on a noise field, so as to solve the problem that the existing image noise reduction scheme has poor applicability and local noise reduction after noise reduction.
  • Technical issues of regional image distortion are to propose an image noise reduction method, device, computer equipment and storage medium based on a noise field, so as to solve the problem that the existing image noise reduction scheme has poor applicability and local noise reduction after noise reduction.
  • the embodiments of the present application provide an image noise reduction method based on a noise field, which adopts the following technical solutions:
  • An image noise reduction method based on noise field comprising:
  • a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, the noise-free image is input into the preset noise distribution model, and the Gauss-Poisson joint noise distribution function of the noise-free image is constructed in the noise distribution model. steps, including:
  • the Gauss-Poisson joint noise distribution function of the noise-free image is constructed based on the pixel information of the noise-free image, the Poisson kernel and the Gaussian kernel.
  • the method further includes:
  • the noise distribution model is iterated based on the recognition error until the model is fitted, and the output fitted noise distribution model is obtained.
  • the steps of iterating the noise distribution model based on the recognition error until the model is fitted, and obtaining the fitted noise distribution model specifically includes:
  • the initial noise distribution model is iteratively updated based on the back-propagation algorithm until the recognition error is less than or equal to the preset error threshold, and the output fitted noise distribution model is obtained;
  • the step of obtaining the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function specifically includes:
  • the Gauss-Poisson joint noise distribution function is randomly sampled based on the pixel information of the noise-free image, and the sampling matrix is obtained;
  • the noise distribution information of the image to be denoised is obtained according to the sampling matrix.
  • the step of obtaining the noise distribution information of the image to be denoised according to the sampling matrix specifically includes:
  • An image matrix of the fourth noise image is obtained, and noise distribution information of the image to be denoised is obtained based on the image matrix of the fourth noise image.
  • Image reconstruction is performed based on the result of the convolution operation to obtain a denoised image.
  • the embodiments of the present application also provide an image noise reduction device based on a noise field, which adopts the following technical solutions:
  • An image noise reduction device based on noise field comprising:
  • an image acquisition module for acquiring the image to be denoised and the noise-free image corresponding to the image to be denoised
  • the function building module is used to input the noise-free image into the preset noise distribution model, and construct the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
  • the noise simulation module is used to obtain the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function
  • the image denoising module is used to import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the denoising image in the denoising model according to the noise distribution information to obtain a denoised image.
  • the embodiment of the present application also provides a computer device, which adopts the following technical solutions:
  • a computer device includes a memory and a processor, wherein computer-readable instructions are stored in the memory, and the processor implements the following noise field-based image noise reduction method when executing the computer-readable instructions:
  • the embodiments of the present application also provide a computer-readable storage medium, which adopts the following technical solutions:
  • a computer-readable storage medium where computer-readable instructions are stored on the computer-readable storage medium, and when the computer-readable instructions are executed by a processor, the following image noise reduction method based on noise field is implemented:
  • the present application discloses an image noise reduction method, device, equipment and storage medium based on noise field, belonging to the technical field of artificial intelligence.
  • the noise distribution model obtains a Gauss-Poisson joint noise distribution function.
  • the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model is based on the noise distribution information.
  • Noise reduction is performed on the image to be denoised to obtain a denoised image.
  • the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
  • FIG. 1 shows an exemplary system architecture diagram to which the present application can be applied
  • FIG. 2 shows a flowchart of an embodiment of a noise field-based image noise reduction method according to the present application
  • FIG. 3 shows a schematic structural diagram of an embodiment of a noise field-based image noise reduction apparatus according to the present application
  • FIG. 4 shows a schematic structural diagram of an embodiment of a computer device according to the present application.
  • the system architecture 100 may include terminal devices 101 , 102 , and 103 , a network 104 and a server 105 .
  • the network 104 is a medium used to provide a communication link between the terminal devices 101 , 102 , 103 and the server 105 .
  • the network 104 may include various connection types, such as wired, wireless communication links, or fiber optic cables, among others.
  • the user can use the terminal devices 101, 102, 103 to interact with the server 105 through the network 104 to receive or send messages and the like.
  • Various communication client applications may be installed on the terminal devices 101 , 102 and 103 , such as web browser applications, shopping applications, search applications, instant messaging tools, email clients, social platform software, and the like.
  • the terminal devices 101, 102, and 103 can be various electronic devices that have a display screen and support web browsing, including but not limited to smart phones, tablet computers, e-book readers, MP3 players (Moving Picture Experts Group Audio Layer III, dynamic Picture Experts Compression Standard Audio Layer 3), MP4 (Moving Picture Experts Group Audio Layer IV, Moving Picture Experts Compression Standard Audio Layer 4) Players, Laptops and Desktops, etc.
  • MP3 players Moving Picture Experts Group Audio Layer III, dynamic Picture Experts Compression Standard Audio Layer 3
  • MP4 Moving Picture Experts Group Audio Layer IV, Moving Picture Experts Compression Standard Audio Layer 4
  • the server 105 may be a server that provides various services, such as a background server that provides support for the pages displayed on the terminal devices 101 , 102 , and 103 .
  • the noise field-based image noise reduction method provided by the embodiments of the present application is generally performed by a server or a terminal device, and accordingly, the noise field-based image noise reduction apparatus is generally set in the server or terminal device.
  • terminal devices, networks and servers in FIG. 1 are merely illustrative. There can be any number of terminal devices, networks and servers according to implementation needs.
  • the image noise reduction method based on noise field includes the following steps:
  • noise-free images refer to images whose image indicators meet the preset requirements.
  • Image indicators such as resolution, exposure, saturation, etc.
  • the image required by the index is determined as a noise-free image
  • the image noise of the image to be denoised is simulated on the noise-free image
  • the noise distribution information is generated based on the simulated image noise
  • the denoised image is to be denoised based on the noise distribution information and the pre-trained noise reduction model. Noise reduction is performed to obtain a denoised image.
  • the electronic device for example, the server or terminal device shown in FIG. 1
  • the noise field-based image noise reduction method runs may receive the image noise reduction instruction through wired connection or wireless connection.
  • the above wireless connection methods may include but are not limited to 3G/4G connection, WiFi connection, Bluetooth connection, WiMAX connection, Zigbee connection, UWB (ultra wideband) connection, and other wireless connection methods currently known or developed in the future .
  • S202 Input the noise-free image into a preset noise distribution model, and construct a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model.
  • image noise is divided into two categories, namely signal-related noise and signal-unrelated noise.
  • Signal-related noise is mainly shot noise, and the noise increases with the increase of the signal, which can be simulated by Poisson distribution.
  • the uncorrelated noise of the signal is mainly random noise.
  • the noise does not increase significantly when the signal increases. It conforms to the Gaussian distribution and can be simulated by the Gaussian distribution. Therefore, in a specific embodiment of the present application, the image noise can be fitted by using a joint Poisson-Gaussian distribution.
  • the noise of the image to be denoised is simulated by constructing a noise distribution model.
  • the noise generated by photon sensing that is, the signal-related noise
  • the noise can be modeled as Poisson noise
  • the rest of the static disturbances can be modeled as Poisson noise.
  • Noise i.e. signal uncorrelated noise
  • the Gauss-Poisson joint noise distribution function can be defined as follows:
  • L is considered to be an ideal noise-free image
  • ⁇ s is considered to be the multiplicative noise related to the signal
  • ⁇ c is considered to be the additive noise independent of the signal
  • ⁇ 2 is the image noise generated by the simulation.
  • S203 Acquire noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function.
  • the Gauss-Poisson joint noise distribution function is randomly sampled based on the pixel information of the noise-free image to obtain a sampling matrix, and the noise distribution information of the image to be denoised is obtained according to the sampling matrix.
  • the input noise-free image is an image of size 512*512
  • the Gauss-Poisson joint noise distribution function needs to be randomly sampled 512*512 times, and the results obtained by random sampling are combined to obtain a 512*512 sampling matrix
  • the sampling matrix is added to the noise-free image, and the camera noise, format conversion noise and compression noise related to the image to be de-noised are added, and finally a real image with noise corresponding to the image to be de-noised is obtained by simulation.
  • the noise distribution information of the image to be denoised is obtained from the noise-free real image and the noise-free image.
  • the noise reduction module here is a U-shaped convolutional network, which is divided into an encoding layer encoder and a decoding layer decoder, and the encoding layer encoder contains a total of three layers of convolutional networks, and the number of convolution channels of each layer of encoding layer encoder is in order are 64, 128, 256.
  • the corresponding decoding layer decoder also includes a three-layer convolutional network, and the number of convolution channels of each layer of the decoding layer decoder is 256, 128, and 64 in turn.
  • the image matrix of the image to be denoised is obtained, the noise distribution matrix is obtained based on the noise distribution information, and the image matrix of the image to be denoised and the noise distribution matrix are matrix spliced to obtain a matrix splicing tensor, where the matrix splicing tensor is A three-dimensional tensor.
  • the matrix splicing tensor is A three-dimensional tensor.
  • the three-dimensional tensors are encoded by the three-layer encoding layer encoder, and then the corresponding encoding results are decoded by the three-layer decoding layer decoder respectively, and finally the decoding results are combined to obtain the output of the noise reduction model. This output is the denoised image.
  • the present application discloses an image noise reduction method based on noise field, which belongs to the technical field of artificial intelligence.
  • the present application obtains a Gauss- Poisson joint noise distribution function, by randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model denoises the image to be denoised according to the noise distribution information. , to get the denoised image.
  • the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
  • a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, the noise-free image is input into the preset noise distribution model, and the Gauss-Poisson joint noise distribution function of the noise-free image is constructed in the noise distribution model. steps, including:
  • the Gauss-Poisson joint noise distribution function of the noise-free image is constructed based on the pixel information of the noise-free image, the Poisson kernel and the Gaussian kernel.
  • a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, the noise-free image is input into the noise distribution model, the pixel information of the noise-free image is obtained through the noise distribution model, and the pixel information of the noise-free image is imported Perform Gaussian operation in the Gaussian kernel to obtain the Gaussian operation result, import the pixel information of the noise-free image into the Poisson kernel to perform the Poisson operation, obtain the Poisson operation result, and construct the Gauss-Poisson joint noise based on the Gaussian operation result and the Poisson operation result. Distribution function.
  • a Gaussian kernel and a Poisson kernel preset in the noise distribution model are used to process the pixel information of the noise-free image respectively, and the Gauss-Poisson joint noise distribution function is constructed and constructed through the obtained processing results,
  • the Gauss-Poisson joint noise distribution function can be used to simulate image noise.
  • the method further includes:
  • the noise distribution model is iterated based on the recognition error until the model is fitted, and the output fitted noise distribution model is obtained.
  • the sample image may be a noise-free image
  • the noise-free image is input into a preset initial noise distribution model to obtain the output result of the initial noise distribution model, where the output result of the initial noise distribution model is a matrix
  • the output result of the initial noise distribution model is a matrix
  • the output result of the initial noise distribution model should also be a matrix of size 512*512.
  • the preset standard result here may be an image containing noise corresponding to the noise-free image. Image.
  • the noise distribution model is iterated by using the back-propagation algorithm until the model is fitted, and the noise distribution model is obtained.
  • the initial noise distribution model can be a prediction model using the ResNet structure
  • ResNet refers to the abbreviation of Residual Network
  • the ResNet prediction model is a classic neural network as the backbone of many computer vision tasks.
  • a Gaussian kernel and a Poisson kernel are added to the ResNet prediction model to construct a noise distribution model, and the noise distribution of the image is simulated by the constructed noise distribution model.
  • a new loss function L needs to be defined here.
  • the new loss function L includes the L1 loss function and the L2 loss function, and the specific definitions are as follows:
  • L1 exhibits the asymmetry of simulated image noise, where i refers to the coordinates of the image matrix, represents the output result of the initial noise distribution model, and ⁇ represents the preset standard result, that is, the image matrix containing the noise image corresponding to the input sample image.
  • I represents a step function, and when the calculation formula of I subscript is less than 0, its value is 1, and when the calculation formula of I subscript is greater than or equal to 0, its value is 0, ⁇ is a manually set constant, usually Can be set between 0 and 0.5 to modulate the value of the loss function.
  • L2 is the guarantee to prevent the distortion of subsequent image synthesis, among which, here and Refers to the horizontal and vertical differentiation of the output results, respectively.
  • the weighting of the above two L1 and L2 constitutes the loss function L of our noise distribution model.
  • the specific form of the loss function L is as follows:
  • ⁇ and ⁇ are weighting coefficients.
  • the initial weighting coefficients of L1 and L2 are set to 0.5 and 0.5 respectively, and then the initial weighting coefficients are continuously adjusted according to the output result of the initial noise distribution model. .
  • the initial noise distribution model is trained through sample images, and the loss function of the initial noise distribution model is constructed, and the output error of the initial noise distribution model is calculated based on the constructed loss function, and the initial noise distribution model is calculated based on the output error. Iterate to obtain a noise distribution model that meets the requirements.
  • the steps of iterating the noise distribution model based on the recognition error until the model is fitted, and obtaining the fitted noise distribution model specifically includes:
  • the initial noise distribution model is iteratively updated based on the back-propagation algorithm until the recognition error is less than or equal to the preset error threshold, and the output fitted noise distribution model is obtained;
  • the backpropagation algorithm that is, the error backpropagation algorithm (Backpropagationalgorithm, BP algorithm) is a learning algorithm suitable for multi-layer neuron networks. It is based on the gradient descent method and is used for the error calculation of deep learning networks. .
  • the input and output relationship of BP network is essentially a mapping relationship: the function completed by a BP neural network with n input and m output is a continuous mapping from n-dimensional Euclidean space to a finite field in m-dimensional Euclidean space. A map is highly nonlinear.
  • the learning process of BP algorithm consists of forward propagation process and back propagation process.
  • the input information is processed layer by layer through the hidden layer through the input layer and transmitted to the output layer, and then transferred to the back propagation, and the partial derivative of the objective function to the weight of each neuron is obtained layer by layer, which constitutes The gradient of the objective function to the weight vector is used as the basis for modifying the weight.
  • the identification error is compared with a preset error threshold, and if the identification error is greater than the preset error threshold, the initial noise distribution model after training is iteratively updated based on the back-propagation algorithm until the identification error is less than or equal to the preset error Up to the threshold, obtain the noise distribution model of the output fitting.
  • the preset error threshold may be set in advance.
  • the initial noise distribution model that has been trained is verified and iterated through the back-propagation algorithm to obtain a noise distribution model that meets the requirements.
  • the step of obtaining the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function specifically includes:
  • the Gauss-Poisson joint noise distribution function is randomly sampled based on the pixel information of the noise-free image, and the sampling matrix is obtained;
  • the noise distribution information of the image to be denoised is obtained according to the sampling matrix.
  • the Gauss-Poisson joint noise distribution function is randomly sampled based on the pixel information of the noise-free image to obtain a sampling matrix, and the noise distribution information of the image to be denoised is obtained according to the sampling matrix.
  • the input noise-free image is an image of size 512*512
  • the Gauss-Poisson joint noise distribution function needs to be randomly sampled 512*512 times, and the results obtained by random sampling are combined to obtain a 512*512 sampling matrix
  • the sampling matrix is added to the noise-free image, and the camera noise, format conversion noise and compression noise related to the image to be de-noised are added, and finally a real image with noise corresponding to the image to be de-noised is obtained by simulation.
  • the noise distribution information of the image to be denoised is obtained from the noise-free real image and the noise-free image.
  • a sampling matrix is obtained by randomly sampling the Gauss-Poisson joint noise distribution function, and the sampling matrix is added to the noise-free image, and the camera noise, format conversion noise and Compress the noise, and finally simulate a real image with noise corresponding to the image to be denoised. Based on the real image with noise and the noise-free image, the noise distribution information of the image to be de-noised can be obtained.
  • the step of obtaining the noise distribution information of the image to be denoised according to the sampling matrix specifically includes:
  • An image matrix of the fourth noise image is obtained, and noise distribution information of the image to be denoised is obtained based on the image matrix of the fourth noise image.
  • the image matrix of the noise-free image is obtained, and the image matrix and the sampling matrix of the noise-free image are fused to obtain the first noise image.
  • the factors process the first noise image in turn to simulate a real image with noise corresponding to the image to be denoised.
  • the process of format conversion from the bayer image of the original camera to the RGB image the format conversion process will generate Certain noise
  • image JPEG compression refers to the compression process before image transmission, and image compression will produce certain noise.
  • the specific operation process of sequentially processing the first noise image according to the above-mentioned influencing factors is as follows:
  • L is considered to be an ideal noise-free image
  • y is the real image with noise
  • f is the camera response function
  • DM refers to the process from the bayer image to the RGB image, that is, the color interpolation process.
  • JPEG is the image compression process. So far, the step of simulating image noise is completed, and a real image containing noise corresponding to the image to be denoised is obtained.
  • the first noise image is processed based on the camera shooting factor, the format conversion factor and the image JPEG compression factor, so that the first noise image obtains the camera noise, the format conversion noise and the compression noise, and finally obtains an image with
  • the noise-containing real image corresponding to the image to be de-noised can be obtained based on the noise-containing real image and the noise-free image to obtain noise distribution information of the image to be de-noised.
  • Image reconstruction is performed based on the result of the convolution operation to obtain a denoised image.
  • the noise distribution matrix is obtained based on the noise distribution information, and the image matrix of the image to be denoised and the noise distribution matrix are matrix spliced to obtain a matrix splicing tensor.
  • the matrix splicing tensor is a three-dimensional condition tensor.
  • the convolution kernel performs the convolution operation on the matrix splicing tensor to obtain the convolution operation result, and fills the convolution operation result into the matrix body of a blank matrix in turn, where the blank matrix is the same size as the image matrix of the image to be denoised, for example They are all 512*512-sized matrices.
  • the above process is equivalent to reconstructing the image, and the denoised image is obtained after reconstruction.
  • the matrix splicing tensor is convolved in a condition-guided manner, and then the image is reconstructed based on the result of the convolution operation to obtain a denoised image.
  • the default is uniform noise distribution for noise reduction, which is applicable in general scenes of natural images, but in mobile scenes, due to shooting angle, light distribution and other reasons, it will cause The noise distribution is not uniform.
  • noise reduction should be carried out based on the noise distribution.
  • the noise reduction should be increased in places with more noise, and the noise reduction should be reduced in places with less noise.
  • the obtained images are sharp, clean, and distorted images.
  • To perform noise reduction based on the noise distribution it is necessary to first estimate the noise distribution of the image, and then perform noise reduction based on the noise distribution.
  • the present application discloses an image noise reduction method, device, equipment and storage medium based on noise field, which belong to the technical field of artificial intelligence.
  • the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained denoising
  • the model denoises the denoised image according to the noise distribution information, and obtains the denoised image.
  • the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
  • the above-mentioned images to be de-noised and noise-free images can also be stored in a node of a blockchain.
  • the blockchain referred to in this application is a new application mode of computer technologies such as distributed data storage, point-to-point transmission, consensus mechanism, and encryption algorithm.
  • Blockchain essentially a decentralized database, is a series of data blocks associated with cryptographic methods. Each data block contains a batch of network transaction information to verify its Validity of information (anti-counterfeiting) and generation of the next block.
  • the blockchain can include the underlying platform of the blockchain, the platform product service layer, and the application service layer.
  • the aforementioned storage medium may be a non-volatile storage medium such as a magnetic disk, an optical disk, a read-only memory (Read-Only Memory, ROM), or a random access memory (Random Access Memory, RAM) or the like.
  • the present application provides an embodiment of an image noise reduction device based on a noise field, and the device embodiment corresponds to the method embodiment shown in FIG. 2 .
  • the device can be specifically applied to various electronic devices.
  • the noise field-based image noise reduction device in this embodiment includes:
  • An image acquisition module 301 configured to acquire an image to be denoised and a noise-free image corresponding to the image to be denoised;
  • a function construction module 302 configured to input the noise-free image into a preset noise distribution model, and construct a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
  • a noise simulation module 303 configured to obtain the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function
  • the image denoising module 304 is used to import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the denoising image in the denoising model according to the noise distribution information to obtain a denoised image.
  • a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, and the function building module 302 specifically includes:
  • an information extraction unit used for inputting the noise-free image into the noise distribution model, and obtaining pixel information of the noise-free image through the noise distribution model
  • the function construction unit is used to construct the Gauss-Poisson joint noise distribution function of the noise-free image based on the pixel information of the noise-free image, the Poisson kernel and the Gaussian kernel.
  • the image noise reduction device based on noise field also includes:
  • a sample acquisition module configured to acquire a sample image from a preset image database, input the sample image into a preset initial noise distribution model, and obtain an output result of the initial noise distribution model
  • the error calculation module is used to construct the loss function of the initial noise distribution model, and based on the output result and the preset standard result, use the loss function of the noise distribution model to perform error calculation to obtain the identification error;
  • the model iteration module is used to iterate the noise distribution model based on the recognition error until the model is fitted, and obtain the output fitted noise distribution model.
  • model iteration module specifically includes:
  • the error comparison unit is used to compare the recognition error with the preset error threshold
  • the model iteration unit is used to iteratively update the initial noise distribution model based on the back-propagation algorithm when the recognition error is greater than the preset error threshold, until the recognition error is less than or equal to the preset error threshold, and obtain the output fitted noise distribution model ;
  • Model output unit for outputting the noise distribution model.
  • the noise simulation module 303 specifically includes:
  • the random sampling unit is used to randomly sample the Gauss-Poisson joint noise distribution function based on the pixel information of the noise-free image to obtain a sampling matrix;
  • the noise simulation unit is used to obtain the noise distribution information of the image to be denoised according to the sampling matrix.
  • noise simulation unit specifically includes:
  • the matrix fusion subunit is used to obtain the image matrix of the noise-free image, and fuse the image matrix and the sampling matrix of the noise-free image to obtain the first noise image;
  • a camera noise simulation subunit configured to obtain a camera response function corresponding to the image to be denoised, and process the first noise image based on the camera response function to obtain a second noise image
  • a format conversion noise simulation subunit used for acquiring format conversion information corresponding to the image to be denoised, and performing color interpolation on the second noise image based on the format conversion information to obtain a third noise image
  • a compression noise simulation subunit used for acquiring compression parameters corresponding to the image to be denoised, and compressing the third noise image based on the compression parameters to obtain a fourth noise image
  • the noise distribution subunit is configured to acquire an image matrix of the fourth noise image, and obtain noise distribution information of the image to be denoised based on the image matrix of the fourth noise image.
  • the image noise reduction module 304 specifically includes:
  • a distribution matrix unit used to obtain a noise distribution matrix based on the noise distribution information
  • the matrix splicing unit is used to perform matrix splicing of the image matrix and the noise distribution matrix of the image to be denoised to obtain a matrix splicing tensor;
  • the convolution operation unit is used to perform the convolution operation on the matrix splicing tensor by using the convolution check of the noise reduction model to obtain the result of the convolution operation;
  • the image reconstruction unit is used for image reconstruction based on the result of the convolution operation to obtain a denoised image.
  • the present application discloses an image noise reduction device based on noise field, which belongs to the technical field of artificial intelligence.
  • the present application obtains a Gauss- Poisson joint noise distribution function, by randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model denoises the image to be denoised according to the noise distribution information. , to get the denoised image.
  • the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
  • FIG. 4 is a block diagram of a basic structure of a computer device according to this embodiment.
  • the computer device 4 includes a memory 41, a processor 42, and a network interface 43 that communicate with each other through a system bus. It should be noted that only the computer device 4 with components 41-43 is shown in the figure, but it should be understood that it is not required to implement all of the shown components, and more or less components may be implemented instead. Among them, those skilled in the art can understand that the computer device here is a device that can automatically perform numerical calculation and/or information processing according to pre-set or stored instructions, and its hardware includes but is not limited to microprocessors, special-purpose Integrated circuit (Application Specific Integrated Circuit, ASIC), programmable gate array (Field-Programmable Gate Array, FPGA), digital processor (Digital Signal Processor, DSP), embedded equipment, etc.
  • ASIC Application Specific Integrated Circuit
  • FPGA Field-Programmable Gate Array
  • DSP Digital Signal Processor
  • the computer equipment may be a desktop computer, a notebook computer, a palmtop computer, a cloud server and other computing equipment.
  • the computer device can perform human-computer interaction with the user through a keyboard, a mouse, a remote control, a touch pad or a voice control device.
  • the memory 41 includes at least one type of readable storage medium, and the readable storage medium includes flash memory, hard disk, multimedia card, card-type memory (for example, SD or DX memory, etc.), random access memory (RAM), static Random Access Memory (SRAM), Read Only Memory (ROM), Electrically Erasable Programmable Read Only Memory (EEPROM), Programmable Read Only Memory (PROM), Magnetic Memory, Magnetic Disk, Optical Disk, etc.
  • the memory 41 may be an internal storage unit of the computer device 4 , such as a hard disk or a memory of the computer device 4 .
  • the memory 41 may also be an external storage device of the computer device 4, such as a plug-in hard disk, a smart memory card (Smart Media Card, SMC), a secure digital (Secure Digital, SD) card, flash memory card (Flash Card), etc.
  • the memory 41 may also include both the internal storage unit of the computer device 4 and its external storage device.
  • the memory 41 is generally used to store the operating system and various application software installed on the computer device 4 , such as computer-readable instructions for an image noise reduction method based on a noise field.
  • the memory 41 can also be used to temporarily store various types of data that have been output or will be output.
  • the processor 42 may be a central processing unit (Central Processing Unit, CPU), a controller, a microcontroller, a microprocessor, or other data processing chips in some embodiments. This processor 42 is typically used to control the overall operation of the computer device 4 . In this embodiment, the processor 42 is configured to execute computer-readable instructions stored in the memory 41 or process data, for example, computer-readable instructions for executing the noise field-based image noise reduction method.
  • CPU Central Processing Unit
  • controller central processing unit
  • microcontroller a microcontroller
  • microprocessor microprocessor
  • This processor 42 is typically used to control the overall operation of the computer device 4 .
  • the processor 42 is configured to execute computer-readable instructions stored in the memory 41 or process data, for example, computer-readable instructions for executing the noise field-based image noise reduction method.
  • the network interface 43 may include a wireless network interface or a wired network interface, and the network interface 43 is generally used to establish a communication connection between the computer device 4 and other electronic devices.
  • the present application discloses a computer device, which belongs to the technical field of artificial intelligence.
  • the present application obtains a Gauss-Poisson joint noise distribution function by inputting a noise-free image with similar or identical content to the image to be denoised into a preset noise distribution model , by randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model denoises the denoised image according to the noise distribution information to obtain a denoised image.
  • the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
  • the present application also provides another implementation manner, that is, to provide a computer-readable storage medium
  • the computer-readable storage medium may be non-volatile or volatile
  • the computer-readable storage medium stores Computer readable instructions executable by at least one processor to cause the at least one processor to perform the steps of the noise field based image noise reduction method as described above.
  • the present application discloses a storage medium, which belongs to the technical field of artificial intelligence.
  • the present application obtains a Gauss-Poisson joint noise distribution function by inputting a noise-free image that is similar or identical to the content of the image to be de-noised into a preset noise distribution model , by randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model denoises the denoised image according to the noise distribution information to obtain a denoised image.
  • the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
  • the method of the above embodiment can be implemented by means of software plus a necessary general hardware platform, and of course can also be implemented by hardware, but in many cases the former is better implementation.
  • the technical solution of the present application can be embodied in the form of a software product in essence or in a part that contributes to the prior art, and the computer software product is stored in a storage medium (such as ROM/RAM, magnetic disk, CD-ROM), including several instructions to make a terminal device (which may be a mobile phone, a computer, a server, an air conditioner, or a network device, etc.) execute the methods described in the various embodiments of this application.
  • a storage medium such as ROM/RAM, magnetic disk, CD-ROM

Landscapes

  • Engineering & Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Theoretical Computer Science (AREA)
  • General Physics & Mathematics (AREA)
  • Data Mining & Analysis (AREA)
  • Molecular Biology (AREA)
  • Biophysics (AREA)
  • Computational Linguistics (AREA)
  • Artificial Intelligence (AREA)
  • Evolutionary Computation (AREA)
  • General Health & Medical Sciences (AREA)
  • Biomedical Technology (AREA)
  • Computing Systems (AREA)
  • General Engineering & Computer Science (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Mathematical Physics (AREA)
  • Software Systems (AREA)
  • Health & Medical Sciences (AREA)
  • Image Processing (AREA)

Abstract

The present application relates to a computer vision technology in the technical field of artificial intelligence, and discloses a noise field-based image noise reduction method and apparatus, a device, and a storage medium. In the present application, the method comprises: obtaining an image to be subjected to noise reduction and a noise-free image corresponding to the image to be subjected to noise reduction; inputting the noise-free image into a preset noise distribution model, and constructing a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model; obtaining, according to the Gauss-Poisson joint noise distribution function, noise distribution information of the image to be subjected to noise reduction; importing the image to be subjected to noise reduction and the noise distribution information to a pre-trained noise reduction model; and performing, in the noise reduction model according to the noise distribution information, noise reduction on the image to be subjected to noise reduction to obtain a denoised image. In addition, the present application further relates to a blockchain technology, and the image to be subjected to noise reduction and the noise-free image can be stored in a blockchain. According to the present application, not only can a clear and clean noise-reduced image be obtained, but the distortion generated during image noise reduction can also be prevented.

Description

基于噪声场的图像降噪方法、装置、设备及存储介质Image noise reduction method, device, device and storage medium based on noise field
本申请要求于2021年2月26日提交中国专利局、申请号为202110219477.9,发明名称为“基于噪声场的图像降噪方法、装置、设备及存储介质”的中国专利申请的优先权,其全部内容通过引用结合在本申请中。This application claims the priority of the Chinese patent application filed on February 26, 2021 with the application number 202110219477.9 and the invention titled "Method, Apparatus, Equipment and Storage Medium for Image Noise Reduction Based on Noise Field", all of which The contents are incorporated herein by reference.
技术领域technical field
本申请涉及人工智能技术领域,具体涉及一种基于噪声场的图像降噪方法、装置、设备及存储介质。The present application relates to the technical field of artificial intelligence, and in particular, to a noise field-based image noise reduction method, device, device, and storage medium.
背景技术Background technique
随着人工智能在金融领域中的广泛使用,在移动端开展金融行为的场景越来越丰富,这些场景中均涉及到较为严格的审批业务,如人脸识别业务,人脸识别业务需要进行客户图像的抽取,出于体验感的考虑,这种留底往往是基于相机拍摄的联系视频中抽帧所得的图片,但在进行图像降噪过程中,发明人意识到目前通过这种方式得到的图片往往质量得不到保障,尤其是移动端本身的移动会造成图像的模糊,这样的图像缺陷对于后续的诸如人脸识别等任务来说是不利的。因此,为了有效提高后续业务诸如人脸识别的准确率,优化图像质量势在必行。With the widespread use of artificial intelligence in the financial field, there are more and more scenarios in which financial behaviors are carried out on mobile terminals. These scenarios involve relatively strict approval services, such as face recognition services, which require customer Image extraction, for the sake of experience, this kind of background is often based on the picture obtained by extracting frames from the contact video shot by the camera, but in the process of image noise reduction, the inventor realized that the current obtained in this way The quality of the pictures is often not guaranteed, especially the movement of the mobile terminal itself will cause the blurring of the images. Such image defects are unfavorable for subsequent tasks such as face recognition. Therefore, in order to effectively improve the accuracy of subsequent services such as face recognition, it is imperative to optimize image quality.
优化图像质量过程的一个必要的步骤是对图像进行降噪处理,通常图像降噪分为两种思路:一是采用人为设计模型的方式来进行图像降噪;二是采用深度学习预估的方式来进行图像降噪。前者往往因为设计的模型只能针对某一种特定情况,因而在大多数情况下适用性不强;近年来兴起的深度学习卷积神经网络的图像降噪方法,虽然具有较好的适应性,但是深度学习预估的方式实际的原理在于使模型对图像整体进行预测,在对图像进行预测之后,根据预测结果对图像整体统一进行降噪处理,但这种整体降噪的方式往往会在局部区域造成图像的畸变。A necessary step in the process of optimizing image quality is to denoise the image. Usually, image denoising is divided into two ideas: one is to use artificially designed models to denoise images; the other is to use deep learning estimation for image noise reduction. The former is often not applicable in most cases because the designed model can only be aimed at a specific situation; the image noise reduction method of deep learning convolutional neural network emerging in recent years, although it has good adaptability, However, the actual principle of the deep learning prediction method is to make the model predict the image as a whole. After the image is predicted, the overall image is denoised according to the prediction result. However, this overall denoising method is often localized Regions cause distortion of the image.
发明内容SUMMARY OF THE INVENTION
本申请实施例的目的在于提出一种基于噪声场的图像降噪方法、装置、计算机设备及存储介质,以解决现有的图像降噪方案存在的适用性不强,且降噪后存在的局部区域图像畸变的技术问题。The purpose of the embodiments of the present application is to propose an image noise reduction method, device, computer equipment and storage medium based on a noise field, so as to solve the problem that the existing image noise reduction scheme has poor applicability and local noise reduction after noise reduction. Technical issues of regional image distortion.
为了解决上述技术问题,本申请实施例提供一种基于噪声场的图像降噪方法,采用了如下所述的技术方案:In order to solve the above technical problems, the embodiments of the present application provide an image noise reduction method based on a noise field, which adopts the following technical solutions:
一种基于噪声场的图像降噪方法,包括:An image noise reduction method based on noise field, comprising:
获取待降噪图像以及与待降噪图像相对应的无噪声图像;Obtain the image to be denoised and the noise-free image corresponding to the image to be denoised;
将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数;Input the noise-free image into the preset noise distribution model, and construct the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
根据高斯-泊松联合噪声分布函数获取待降噪图像的噪声分布信息;Obtain the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
将待降噪图像和噪声分布信息导入到预先训练好的降噪模型,根据噪声分布信息在以降噪模型中对待降噪图像进行降噪,得到去噪图像。Import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain a denoised image.
进一步地,噪声分布模型中预设有一个高斯核和一个泊松核,将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数的步骤,具体包括:Further, a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, the noise-free image is input into the preset noise distribution model, and the Gauss-Poisson joint noise distribution function of the noise-free image is constructed in the noise distribution model. steps, including:
将无噪声图像输入到噪声分布模型中,通过噪声分布模型获取无噪声图像的像素信息;Input the noise-free image into the noise distribution model, and obtain the pixel information of the noise-free image through the noise distribution model;
基于无噪声图像的像素信息、泊松核以及高斯核构建无噪声图像的高斯-泊松联合噪声分布函数。The Gauss-Poisson joint noise distribution function of the noise-free image is constructed based on the pixel information of the noise-free image, the Poisson kernel and the Gaussian kernel.
进一步地,在将无噪声图像输入到噪声分布模型中,通过噪声分布模型获取无噪声图像的像素信息的步骤之前,还包括:Further, before the step of inputting the noise-free image into the noise distribution model, and obtaining the pixel information of the noise-free image through the noise distribution model, the method further includes:
从预设的图像数据库中获取样本图像,将样本图像输入到预设的初始噪声分布模型,获取初始噪声分布模型的输出结果;Obtain a sample image from a preset image database, input the sample image into a preset initial noise distribution model, and obtain the output result of the initial noise distribution model;
构建初始噪声分布模型的损失函数,基于输出结果和预设标准结果,使用噪声分布模型的损失函数进行误差计算,获取识别误差;Construct the loss function of the initial noise distribution model, and use the loss function of the noise distribution model to calculate the error based on the output results and the preset standard results to obtain the recognition error;
基于识别误差对噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型。The noise distribution model is iterated based on the recognition error until the model is fitted, and the output fitted noise distribution model is obtained.
进一步地,基于识别误差对噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型的步骤,具体包括:Further, the steps of iterating the noise distribution model based on the recognition error until the model is fitted, and obtaining the fitted noise distribution model, specifically includes:
将识别误差与预设误差阈值进行比对;Compare the recognition error with the preset error threshold;
若识别误差大于预设误差阈值,则基于反向传播算法对初始噪声分布模型进行迭代更新,直至识别误差小于或等于预设误差阈值为止,得到输出拟合的噪声分布模型;If the recognition error is greater than the preset error threshold, the initial noise distribution model is iteratively updated based on the back-propagation algorithm until the recognition error is less than or equal to the preset error threshold, and the output fitted noise distribution model is obtained;
输出噪声分布模型。Output noise distribution model.
进一步地,根据高斯-泊松联合噪声分布函数获取待降噪图像的噪声分布信息的步骤,具体包括:Further, the step of obtaining the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function specifically includes:
基于无噪声图像的像素信息对高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵;The Gauss-Poisson joint noise distribution function is randomly sampled based on the pixel information of the noise-free image, and the sampling matrix is obtained;
根据抽样矩阵得到待降噪图像的噪声分布信息。The noise distribution information of the image to be denoised is obtained according to the sampling matrix.
进一步地,根据抽样矩阵得到待降噪图像的噪声分布信息的步骤,具体包括:Further, the step of obtaining the noise distribution information of the image to be denoised according to the sampling matrix specifically includes:
获取无噪声图像的图像矩阵,融合无噪声图像的图像矩阵和抽样矩阵,得到第一噪声图像;Obtain the image matrix of the noise-free image, and fuse the image matrix and the sampling matrix of the noise-free image to obtain the first noise image;
获取待降噪图像对应的相机响应函数,并基于相机响应函数对第一噪声图像处理,得到第二噪声图像;obtaining a camera response function corresponding to the image to be denoised, and processing the first noise image based on the camera response function to obtain a second noise image;
获取待降噪图像对应的格式转化信息,并基于格式转化信息对第二噪声图像进行彩色插值,得到第三噪声图像;acquiring format conversion information corresponding to the image to be denoised, and performing color interpolation on the second noise image based on the format conversion information to obtain a third noise image;
获取待降噪图像对应的压缩参数,并基于压缩参数对第三噪声图像进行压缩,得到第四噪声图像;obtaining compression parameters corresponding to the image to be denoised, and compressing the third noise image based on the compression parameters to obtain a fourth noise image;
获取第四噪声图像的的图像矩阵,并基于第四噪声图像的的图像矩阵得到待降噪图像的噪声分布信息。An image matrix of the fourth noise image is obtained, and noise distribution information of the image to be denoised is obtained based on the image matrix of the fourth noise image.
进一步地,将待降噪图像和噪声分布信息导入到预先训练好的降噪模型,根据噪声分布信息在以降噪模型中对待降噪图像进行降噪,得到去噪图像的步骤,具体包括:Further, import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain the denoised image, specifically including:
基于噪声分布信息获取噪声分布矩阵;Obtain the noise distribution matrix based on the noise distribution information;
对待降噪图像的图像矩阵和噪声分布矩阵进行矩阵拼接,得到矩阵拼接张量;Perform matrix splicing on the image matrix and noise distribution matrix of the image to be denoised to obtain a matrix splicing tensor;
利用降噪模型的卷积核对矩阵拼接张量进行卷积运算,得到卷积运算结果;Use the convolution check of the noise reduction model to perform the convolution operation on the matrix splicing tensor to obtain the result of the convolution operation;
基于卷积运算结果进行图像重建,得到去噪图像。Image reconstruction is performed based on the result of the convolution operation to obtain a denoised image.
为了解决上述技术问题,本申请实施例还提供一种基于噪声场的图像降噪装置,采用了如下所述的技术方案:In order to solve the above technical problems, the embodiments of the present application also provide an image noise reduction device based on a noise field, which adopts the following technical solutions:
一种基于噪声场的图像降噪装置,包括:An image noise reduction device based on noise field, comprising:
图像获取模块,用于获取待降噪图像以及与待降噪图像相对应的无噪声图像;an image acquisition module for acquiring the image to be denoised and the noise-free image corresponding to the image to be denoised;
函数构建模块,用于将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数;The function building module is used to input the noise-free image into the preset noise distribution model, and construct the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
噪声模拟模块,用于根据高斯-泊松联合噪声分布函数获取待降噪图像的噪声分布信息;The noise simulation module is used to obtain the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
图像降噪模块,用于将待降噪图像和噪声分布信息导入到预先训练好的降噪模型,根据噪声分布信息在以降噪模型中对待降噪图像进行降噪,得到去噪图像。The image denoising module is used to import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the denoising image in the denoising model according to the noise distribution information to obtain a denoised image.
为了解决上述技术问题,本申请实施例还提供一种计算机设备,采用了如下所述的技术方案:In order to solve the above-mentioned technical problems, the embodiment of the present application also provides a computer device, which adopts the following technical solutions:
一种计算机设备,包括存储器和处理器,存储器中存储有计算机可读指令,处理器执行计算机可读指令时实现如下的基于噪声场的图像降噪方法的:A computer device includes a memory and a processor, wherein computer-readable instructions are stored in the memory, and the processor implements the following noise field-based image noise reduction method when executing the computer-readable instructions:
为了解决上述技术问题,本申请实施例还提供一种计算机可读存储介质,采用了如下所述的技术方案:In order to solve the above technical problems, the embodiments of the present application also provide a computer-readable storage medium, which adopts the following technical solutions:
获取待降噪图像以及与待降噪图像相对应的无噪声图像;Obtain the image to be denoised and the noise-free image corresponding to the image to be denoised;
将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数;Input the noise-free image into the preset noise distribution model, and construct the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
根据高斯-泊松联合噪声分布函数获取待降噪图像的噪声分布信息;Obtain the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
将待降噪图像和噪声分布信息导入到预先训练好的降噪模型,根据噪声分布信息在以降噪模型中对待降噪图像进行降噪,得到去噪图像。Import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain a denoised image.
一种计算机可读存储介质,计算机可读存储介质上存储有计算机可读指令,计算机可读指令被处理器执行时实现如下的基于噪声场的图像降噪方法:A computer-readable storage medium, where computer-readable instructions are stored on the computer-readable storage medium, and when the computer-readable instructions are executed by a processor, the following image noise reduction method based on noise field is implemented:
获取待降噪图像以及与待降噪图像相对应的无噪声图像;Obtain the image to be denoised and the noise-free image corresponding to the image to be denoised;
将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数;Input the noise-free image into the preset noise distribution model, and construct the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
根据高斯-泊松联合噪声分布函数获取待降噪图像的噪声分布信息;Obtain the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
将待降噪图像和噪声分布信息导入到预先训练好的降噪模型,根据噪声分布信息在以降噪模型中对待降噪图像进行降噪,得到去噪图像。Import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain a denoised image.
与现有技术相比,本申请实施例主要有以下有益效果:Compared with the prior art, the embodiments of the present application mainly have the following beneficial effects:
本申请公开了一种基于噪声场的图像降噪方法、装置、设备及存储介质,属于人工智能技术领域,本申请通过将与待降噪图像内容相近或者相同的无噪声图像输入到预设的噪声分布模型得到一个高斯-泊松联合噪声分布函数,通过对高斯-泊松联合噪声分布函数进行随机抽样,模拟得到待降噪图像的噪声分布信息,然后预先训练的降噪模型根据噪声分布信息对待降噪图像进行降噪,得到去噪图像。相比于现有的对图像整体统一进行降噪方案,本申请基于噪声分布来进行降噪,即针对噪声越多的地方加大降噪力度,而针对噪声较少的地方减小降噪力度,这样不仅能够获得清晰、干净的降噪图像,而且能够防止图像降噪过程中产生的畸变。The present application discloses an image noise reduction method, device, equipment and storage medium based on noise field, belonging to the technical field of artificial intelligence. The noise distribution model obtains a Gauss-Poisson joint noise distribution function. By randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model is based on the noise distribution information. Noise reduction is performed on the image to be denoised to obtain a denoised image. Compared with the existing unified denoising scheme for the whole image, the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
附图说明Description of drawings
为了更清楚地说明本申请中的方案,下面将对本申请实施例描述中所需要使用的附图作一个简单介绍,显而易见地,下面描述中的附图是本申请的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。In order to illustrate the solutions in the present application more clearly, the following will briefly introduce the accompanying drawings used in the description of the embodiments of the present application. For those of ordinary skill, other drawings can also be obtained from these drawings without any creative effort.
图1示出了本申请可以应用于其中的示例性系统架构图;FIG. 1 shows an exemplary system architecture diagram to which the present application can be applied;
图2示出了根据本申请的基于噪声场的图像降噪方法的一个实施例的流程图;FIG. 2 shows a flowchart of an embodiment of a noise field-based image noise reduction method according to the present application;
图3示出了根据本申请的基于噪声场的图像降噪装置的一个实施例的结构示意图;FIG. 3 shows a schematic structural diagram of an embodiment of a noise field-based image noise reduction apparatus according to the present application;
图4示出了根据本申请的计算机设备的一个实施例的结构示意图。FIG. 4 shows a schematic structural diagram of an embodiment of a computer device according to the present application.
具体实施方式Detailed ways
除非另有定义,本文所使用的所有的技术和科学术语与属于本申请的技术领域的技术人员通常理解的含义相同;本文中在申请的说明书中所使用的术语只是为了描述具体的实施例的目的,不是旨在于限制本申请;本申请的说明书和权利要求书及上述附图说明中的术语“包括”和“具有”以及它们的任何变形,意图在于覆盖不排他的包含。本申请的说明书和权利要求书或上述附图中的术语“第一”、“第二”等是用于区别不同对象,而不是用于描述特定顺序。Unless otherwise defined, all technical and scientific terms used herein have the same meaning as commonly understood by one of ordinary skill in the technical field of this application; the terms used herein in the specification of the application are for the purpose of describing specific embodiments only It is not intended to limit the application; the terms "comprising" and "having" and any variations thereof in the description and claims of this application and the above description of the drawings are intended to cover non-exclusive inclusion. The terms "first", "second" and the like in the description and claims of the present application or the above drawings are used to distinguish different objects, rather than to describe a specific order.
在本文中提及“实施例”意味着,结合实施例描述的特定特征、结构或特性可以包含在本申请的至少一个实施例中。在说明书中的各个位置出现该短语并不一定均是指相同的实施例,也不是与其它实施例互斥的独立的或备选的实施例。本领域技术人员显式地和隐式地理解的是,本文所描述的实施例可以与其它实施例相结合。Reference herein to an "embodiment" means that a particular feature, structure, or characteristic described in connection with the embodiment can be included in at least one embodiment of the present application. The appearances of the phrase in various places in the specification are not necessarily all referring to the same embodiment, nor a separate or alternative embodiment that is mutually exclusive of other embodiments. It is explicitly and implicitly understood by those skilled in the art that the embodiments described herein may be combined with other embodiments.
为了使本技术领域的人员更好地理解本申请方案,下面将结合附图,对本申请实施例中的技术方案进行清楚、完整地描述。In order to make those skilled in the art better understand the solutions of the present application, the technical solutions in the embodiments of the present application will be described clearly and completely below with reference to the accompanying drawings.
如图1所示,系统架构100可以包括终端设备101、102、103,网络104和服务器105。网络104用以在终端设备101、102、103和服务器105之间提供通信链路的介质。网络104可以包括各种连接类型,例如有线、无线通信链路或者光纤电缆等等。As shown in FIG. 1 , the system architecture 100 may include terminal devices 101 , 102 , and 103 , a network 104 and a server 105 . The network 104 is a medium used to provide a communication link between the terminal devices 101 , 102 , 103 and the server 105 . The network 104 may include various connection types, such as wired, wireless communication links, or fiber optic cables, among others.
用户可以使用终端设备101、102、103通过网络104与服务器105交互,以接收或发送消息等。终端设备101、102、103上可以安装有各种通讯客户端应用,例如网页浏览器应用、购物类应用、搜索类应用、即时通信工具、邮箱客户端、社交平台软件等。The user can use the terminal devices 101, 102, 103 to interact with the server 105 through the network 104 to receive or send messages and the like. Various communication client applications may be installed on the terminal devices 101 , 102 and 103 , such as web browser applications, shopping applications, search applications, instant messaging tools, email clients, social platform software, and the like.
终端设备101、102、103可以是具有显示屏并且支持网页浏览的各种电子设备,包括但不限于智能手机、平板电脑、电子书阅读器、MP3播放器(Moving Picture Experts Group Audio Layer III,动态影像专家压缩标准音频层面3)、MP4(Moving Picture Experts Group Audio Layer IV,动态影像专家压缩标准音频层面4)播放器、膝上型便携计算机和台式计算机等等。The terminal devices 101, 102, and 103 can be various electronic devices that have a display screen and support web browsing, including but not limited to smart phones, tablet computers, e-book readers, MP3 players (Moving Picture Experts Group Audio Layer III, dynamic Picture Experts Compression Standard Audio Layer 3), MP4 (Moving Picture Experts Group Audio Layer IV, Moving Picture Experts Compression Standard Audio Layer 4) Players, Laptops and Desktops, etc.
服务器105可以是提供各种服务的服务器,例如对终端设备101、102、103上显示的页面提供支持的后台服务器。The server 105 may be a server that provides various services, such as a background server that provides support for the pages displayed on the terminal devices 101 , 102 , and 103 .
需要说明的是,本申请实施例所提供的基于噪声场的图像降噪方法一般由服务器或终端设备执行,相应地,基于噪声场的图像降噪装置一般设置于服务器或终端设备中。It should be noted that the noise field-based image noise reduction method provided by the embodiments of the present application is generally performed by a server or a terminal device, and accordingly, the noise field-based image noise reduction apparatus is generally set in the server or terminal device.
应该理解,图1中的终端设备、网络和服务器的数目仅仅是示意性的。根据实现需要,可以具有任意数目的终端设备、网络和服务器。It should be understood that the numbers of terminal devices, networks and servers in FIG. 1 are merely illustrative. There can be any number of terminal devices, networks and servers according to implementation needs.
继续参考图2,示出了根据本申请的基于噪声场的图像降噪的方法的一个实施例的流程图。所述的基于噪声场的图像降噪方法,包括以下步骤:Continuing to refer to FIG. 2 , a flowchart of an embodiment of a method for noise field-based image denoising according to the present application is shown. The image noise reduction method based on noise field includes the following steps:
S201,获取待降噪图像以及与待降噪图像相对应的无噪声图像;S201, acquiring an image to be denoised and a noise-free image corresponding to the image to be denoised;
具体的,接收图像降噪指令,获取待降噪图像以及与降噪图像相对应的无噪声图像,其中,无噪声图像与待降噪图像的图像内容相同或者相似,例如,在同一场景下用同一相机拍摄的图像,无噪声图像图像指的是图像指标符合预设要求的图像,图像指标例如分辨率、曝光度、饱和度等等,可以预先设定图像指标的要求,将符合预设图像指标要求的图像确定为无噪声图像,在无噪声图像上模拟待降噪图像的图像噪声,并基于模拟的图像噪声生成噪声分布信息,基于噪声分布信息和预先训练的降噪模型对待降噪图像进行降噪,得到去噪图像。Specifically, an image noise reduction instruction is received, and an image to be denoised and a noise-free image corresponding to the noise-reduced image are obtained, wherein the noise-free image is the same or similar to the image content of the image to be de-noised, for example, in the same scene, using Images captured by the same camera, noise-free images refer to images whose image indicators meet the preset requirements. Image indicators such as resolution, exposure, saturation, etc., can be preset to meet the requirements of the image indicators and will meet the preset image requirements The image required by the index is determined as a noise-free image, the image noise of the image to be denoised is simulated on the noise-free image, and the noise distribution information is generated based on the simulated image noise, and the denoised image is to be denoised based on the noise distribution information and the pre-trained noise reduction model. Noise reduction is performed to obtain a denoised image.
在本实施例中,基于噪声场的图像降噪方法运行于其上的电子设备(例如图1所示的服务器或终端设备)可以通过有线连接方式或者无线连接方式接收图像降噪指令。需要指出的是,上述无线连接方式可以包括但不限于3G/4G连接、WiFi连接、蓝牙连接、WiMAX 连接、Zigbee连接、UWB(ultra wideband)连接、以及其他现在已知或将来开发的无线连接方式。In this embodiment, the electronic device (for example, the server or terminal device shown in FIG. 1 ) on which the noise field-based image noise reduction method runs may receive the image noise reduction instruction through wired connection or wireless connection. It should be pointed out that the above wireless connection methods may include but are not limited to 3G/4G connection, WiFi connection, Bluetooth connection, WiMAX connection, Zigbee connection, UWB (ultra wideband) connection, and other wireless connection methods currently known or developed in the future .
S202,将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数。S202: Input the noise-free image into a preset noise distribution model, and construct a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model.
其中,图像噪声分为两类,分别为信号相关噪声与信号不相关噪声。信号相关噪声主要是散粒噪声,噪声随信号的增大而变大,可以使用泊松分布来模拟。信号不相关噪声主要是随机噪声,信号增大噪声没有明显变大,符合高斯分布,可以使用高斯分布来模拟。因此,在本申请具体的实施例中,图像噪声可以使用泊松-高斯联合分布来进行拟合。Among them, image noise is divided into two categories, namely signal-related noise and signal-unrelated noise. Signal-related noise is mainly shot noise, and the noise increases with the increase of the signal, which can be simulated by Poisson distribution. The uncorrelated noise of the signal is mainly random noise. The noise does not increase significantly when the signal increases. It conforms to the Gaussian distribution and can be simulated by the Gaussian distribution. Therefore, in a specific embodiment of the present application, the image noise can be fitted by using a joint Poisson-Gaussian distribution.
在本申请具体的实施例中,通过构建噪声分布模型来模拟待降噪图像的噪声,根据上述理论光子传感产生的噪声,即信号相关噪声可以建模为泊松噪声,而其余的静止扰动噪声,即信号不相关噪声可以建模为高斯分布。将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数,通过高斯-泊松联合噪声分布函数模拟生成待降噪图像的图像噪声,高斯-泊松联合噪声分布函数可做如下定义:In a specific embodiment of the present application, the noise of the image to be denoised is simulated by constructing a noise distribution model. According to the above theoretical photon sensing, the noise generated by photon sensing, that is, the signal-related noise, can be modeled as Poisson noise, while the rest of the static disturbances can be modeled as Poisson noise. Noise, i.e. signal uncorrelated noise, can be modeled as a Gaussian distribution. Input the noise-free image into the preset noise distribution model, and construct the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model, and generate the image noise of the image to be denoised by simulating the Gauss-Poisson joint noise distribution function. , the Gauss-Poisson joint noise distribution function can be defined as follows:
Figure PCTCN2021090430-appb-000001
Figure PCTCN2021090430-appb-000001
其中,L认为是理想的无噪声图像,σ s认为是与信号相关的乘性噪声,而σ c认为是与信号无关的加性噪声,σ 2就是模拟生成的图像噪声。 Among them, L is considered to be an ideal noise-free image, σ s is considered to be the multiplicative noise related to the signal, σ c is considered to be the additive noise independent of the signal, and σ 2 is the image noise generated by the simulation.
S203,根据高斯-泊松联合噪声分布函数获取待降噪图像的噪声分布信息。S203: Acquire noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function.
具体的,基于无噪声图像的像素信息对高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵,根据抽样矩阵得到待降噪图像的噪声分布信息。例如,输入的无噪声图像为512*512大小的图像,则需要对高斯-泊松联合噪声分布函数进行512*512次随机抽样,将随机抽样获得的结果进行组合,得到512*512抽样矩阵,将抽样矩阵添加到无噪声图像中,并添加与待降噪图像相关的相机噪声、格式转化噪声以及压缩噪声,最终模拟得到一张与待降噪图像相对应的含有噪声的真实图像,基于含噪声的真实图像和无噪声图像得到待降噪图像的噪声分布信息。Specifically, the Gauss-Poisson joint noise distribution function is randomly sampled based on the pixel information of the noise-free image to obtain a sampling matrix, and the noise distribution information of the image to be denoised is obtained according to the sampling matrix. For example, if the input noise-free image is an image of size 512*512, the Gauss-Poisson joint noise distribution function needs to be randomly sampled 512*512 times, and the results obtained by random sampling are combined to obtain a 512*512 sampling matrix, The sampling matrix is added to the noise-free image, and the camera noise, format conversion noise and compression noise related to the image to be de-noised are added, and finally a real image with noise corresponding to the image to be de-noised is obtained by simulation. The noise distribution information of the image to be denoised is obtained from the noise-free real image and the noise-free image.
S204,将待降噪图像和噪声分布信息导入到预先训练好的降噪模型,根据噪声分布信息在以降噪模型中对待降噪图像进行降噪,得到去噪图像。S204 , import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain a denoised image.
其中,这里的降噪模块是一个U型的卷积网络,分为编码层ecoder和解码层decoder,且编码层ecoder共包含三层卷积网络,每一层编码层ecoder的卷积通道数依次是64、128、256。对应的解码层decoder也是包含三层卷积网络,每一层解码层decoder的卷积通道数依次是256、128、64。Among them, the noise reduction module here is a U-shaped convolutional network, which is divided into an encoding layer encoder and a decoding layer decoder, and the encoding layer encoder contains a total of three layers of convolutional networks, and the number of convolution channels of each layer of encoding layer encoder is in order are 64, 128, 256. The corresponding decoding layer decoder also includes a three-layer convolutional network, and the number of convolution channels of each layer of the decoding layer decoder is 256, 128, and 64 in turn.
具体的,获取待降噪图像的图像矩阵,并基于噪声分布信息获取噪声分布矩阵,对待降噪图像的图像矩阵和噪声分布矩阵进行矩阵拼接,得到矩阵拼接张量,其中,矩阵拼接张量是一个三维张量。在获得矩阵拼接张量后,通过三层编码层ecoder分别对三维张量进行编码,然后通过三层解码层decoder分别对相应的编码结果进行解码,最后组合解码结果,得到降噪模型的输出,该输出即为去噪图像。Specifically, the image matrix of the image to be denoised is obtained, the noise distribution matrix is obtained based on the noise distribution information, and the image matrix of the image to be denoised and the noise distribution matrix are matrix spliced to obtain a matrix splicing tensor, where the matrix splicing tensor is A three-dimensional tensor. After obtaining the matrix splicing tensor, the three-dimensional tensors are encoded by the three-layer encoding layer encoder, and then the corresponding encoding results are decoded by the three-layer decoding layer decoder respectively, and finally the decoding results are combined to obtain the output of the noise reduction model. This output is the denoised image.
本申请公开了一种基于噪声场的图像降噪方法,属于人工智能技术领域,本申请通过将与待降噪图像内容相近或者相同的无噪声图像输入到预设的噪声分布模型得到一个高斯-泊松联合噪声分布函数,通过对高斯-泊松联合噪声分布函数进行随机抽样,模拟得到待降噪图像的噪声分布信息,然后预先训练的降噪模型根据噪声分布信息对待降噪图像进行降噪,得到去噪图像。相比于现有的对图像整体统一进行降噪方案,本申请基于噪声分布来进行降噪,即针对噪声越多的地方加大降噪力度,而针对噪声较少的地方减小降噪力度,这样不仅能够获得清晰、干净的降噪图像,而且能够防止图像降噪过程中产生的畸变。The present application discloses an image noise reduction method based on noise field, which belongs to the technical field of artificial intelligence. The present application obtains a Gauss- Poisson joint noise distribution function, by randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model denoises the image to be denoised according to the noise distribution information. , to get the denoised image. Compared with the existing unified denoising scheme for the whole image, the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
进一步地,噪声分布模型中预设有一个高斯核和一个泊松核,将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数的步骤,具体包括:Further, a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, the noise-free image is input into the preset noise distribution model, and the Gauss-Poisson joint noise distribution function of the noise-free image is constructed in the noise distribution model. steps, including:
将无噪声图像输入到噪声分布模型中,通过噪声分布模型获取无噪声图像的像素信息;Input the noise-free image into the noise distribution model, and obtain the pixel information of the noise-free image through the noise distribution model;
基于无噪声图像的像素信息、泊松核以及高斯核构建无噪声图像的高斯-泊松联合噪声分布函数。The Gauss-Poisson joint noise distribution function of the noise-free image is constructed based on the pixel information of the noise-free image, the Poisson kernel and the Gaussian kernel.
具体的,噪声分布模型中预设有一个高斯核和一个泊松核,将无噪声图像输入到噪声分布模型中,通过噪声分布模型获取无噪声图像的像素信息,将无噪声图像的像素信息导入高斯核中进行高斯运算,得到高斯运算结果,将无噪声图像的像素信息导入泊松核中进行泊松运算,得到泊松运算结果,基于高斯运算结果和泊松运算结果构建高斯-泊松联合噪声分布函数。Specifically, a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, the noise-free image is input into the noise distribution model, the pixel information of the noise-free image is obtained through the noise distribution model, and the pixel information of the noise-free image is imported Perform Gaussian operation in the Gaussian kernel to obtain the Gaussian operation result, import the pixel information of the noise-free image into the Poisson kernel to perform the Poisson operation, obtain the Poisson operation result, and construct the Gauss-Poisson joint noise based on the Gaussian operation result and the Poisson operation result. Distribution function.
在上述实施例中,通过利用噪声分布模型中预设有一个高斯核和一个泊松核分别对无噪声图像的像素信息进行处理,通过得到的处理结果构建构建高斯-泊松联合噪声分布函数,高斯-泊松联合噪声分布函数可用于模拟图像噪声。In the above embodiment, a Gaussian kernel and a Poisson kernel preset in the noise distribution model are used to process the pixel information of the noise-free image respectively, and the Gauss-Poisson joint noise distribution function is constructed and constructed through the obtained processing results, The Gauss-Poisson joint noise distribution function can be used to simulate image noise.
进一步地,在将无噪声图像输入到噪声分布模型中,通过噪声分布模型获取无噪声图像的像素信息的步骤之前,还包括:Further, before the step of inputting the noise-free image into the noise distribution model, and obtaining the pixel information of the noise-free image through the noise distribution model, the method further includes:
从预设的图像数据库中获取样本图像,将样本图像输入到预设的初始噪声分布模型,获取初始噪声分布模型的输出结果;Obtain a sample image from a preset image database, input the sample image into a preset initial noise distribution model, and obtain the output result of the initial noise distribution model;
构建初始噪声分布模型的损失函数,基于输出结果和预设标准结果,使用噪声分布模型的损失函数进行误差计算,获取识别误差;Construct the loss function of the initial noise distribution model, and use the loss function of the noise distribution model to calculate the error based on the output results and the preset standard results to obtain the recognition error;
基于识别误差对噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型。The noise distribution model is iterated based on the recognition error until the model is fitted, and the output fitted noise distribution model is obtained.
具体的,样本图像可以是一张无噪声的图像,将该无噪声的图像输入到预设的初始噪声分布模型,获取初始噪声分布模型的输出结果,这里初始噪声分布模型的输出结果是以矩阵形式表示吗,例如,样本图像为512*512大小的图像,那么初始噪声分布模型的输出结果也应为512*512大小的矩阵。基于初始噪声分布模型的输出结果和预设标准结果,使用噪声分布模型的损失函数进行误差计算,获取识别误差,这里的预设标准结果可以是与将该无噪声的图像对应的一张含有噪声的图像。基于上述识别误差使用反向传播算法对噪声分布模型进行迭代,直至模型拟合,得到噪声分布模型。Specifically, the sample image may be a noise-free image, and the noise-free image is input into a preset initial noise distribution model to obtain the output result of the initial noise distribution model, where the output result of the initial noise distribution model is a matrix For example, if the sample image is an image of size 512*512, then the output result of the initial noise distribution model should also be a matrix of size 512*512. Based on the output result of the initial noise distribution model and the preset standard result, use the loss function of the noise distribution model to perform error calculation to obtain the recognition error. The preset standard result here may be an image containing noise corresponding to the noise-free image. Image. Based on the above identification error, the noise distribution model is iterated by using the back-propagation algorithm until the model is fitted, and the noise distribution model is obtained.
其中,初始噪声分布模型可以是采用ResNet结构的预测模型,ResNet指的是残差网络(Residual Network)的缩写,ResNet预测模型是一种作为许多计算机视觉任务主干的经典神经网络,本申请通过在ResNet预测模型中添加一个高斯核和一个泊松核构建出一个噪声分布模型,通过构建出的噪声分布模型模拟图像的噪声分布。在本申请具体的实施例中,为了能真实刻画出待降噪图像噪声,这里需要定义一种新的损失函数L,新的损失函数L包括L1损失函数和L2损失函数,具体定义如下:Among them, the initial noise distribution model can be a prediction model using the ResNet structure, ResNet refers to the abbreviation of Residual Network, and the ResNet prediction model is a classic neural network as the backbone of many computer vision tasks. A Gaussian kernel and a Poisson kernel are added to the ResNet prediction model to construct a noise distribution model, and the noise distribution of the image is simulated by the constructed noise distribution model. In the specific embodiment of this application, in order to truly describe the image noise to be denoised, a new loss function L needs to be defined here. The new loss function L includes the L1 loss function and the L2 loss function, and the specific definitions are as follows:
Figure PCTCN2021090430-appb-000002
Figure PCTCN2021090430-appb-000002
Figure PCTCN2021090430-appb-000003
Figure PCTCN2021090430-appb-000003
这里这样设计L1的优点在于展现出模拟生成图像噪声的不对称性,其中,i是指图像矩阵的坐标,
Figure PCTCN2021090430-appb-000004
表示初始噪声分布模型的输出结果,σ表示预设标准结果,即输入的样本图像对应的含有噪声图像的图像矩阵。I表示的是一个阶跃函数,且当I下标的计算公 式小于0时,其值为1,当I下标的计算公式大于等于0时,其值为0,α是一个手动设置的常数,通常可以设置为0到0.5之间,用于调制损失函数的值。
The advantage of designing L1 like this here is that it exhibits the asymmetry of simulated image noise, where i refers to the coordinates of the image matrix,
Figure PCTCN2021090430-appb-000004
represents the output result of the initial noise distribution model, and σ represents the preset standard result, that is, the image matrix containing the noise image corresponding to the input sample image. I represents a step function, and when the calculation formula of I subscript is less than 0, its value is 1, and when the calculation formula of I subscript is greater than or equal to 0, its value is 0, α is a manually set constant, usually Can be set between 0 and 0.5 to modulate the value of the loss function.
另外,为了防止训练过程中,随机性过强而导致预测噪声的连续性过差,根据噪声理论,图像结构具有完整的连续性,噪声也应当具有连续性,而不是在某些区域发生突变,因此需要设置L2损失函数,L2是防止后续图像合成产生畸变的保证,其中,这里
Figure PCTCN2021090430-appb-000005
Figure PCTCN2021090430-appb-000006
分别是指对输出结果进行水平方向和垂直方向的微分。通过这上面两者L1与L2的加权构成了我们的噪声分布模型的损失函数L,损失函数L的具体形式如下:
In addition, in order to prevent the randomness from being too strong during the training process, the continuity of the prediction noise is too poor. According to the noise theory, the image structure has complete continuity, and the noise should also have continuity, rather than abrupt changes in some areas. Therefore, it is necessary to set the L2 loss function. L2 is the guarantee to prevent the distortion of subsequent image synthesis, among which, here
Figure PCTCN2021090430-appb-000005
and
Figure PCTCN2021090430-appb-000006
Refers to the horizontal and vertical differentiation of the output results, respectively. The weighting of the above two L1 and L2 constitutes the loss function L of our noise distribution model. The specific form of the loss function L is as follows:
L=ωL 1+βL 2 L=ωL 1 +βL 2
其中,ω和β为加权系数,在本申请具体的实施例中,L1和L2的初始加权系数分别设定为0.5和0.5,然后再根据初始噪声分布模型的输出结果持续对初始加权系数进行调整。Among them, ω and β are weighting coefficients. In the specific embodiment of the present application, the initial weighting coefficients of L1 and L2 are set to 0.5 and 0.5 respectively, and then the initial weighting coefficients are continuously adjusted according to the output result of the initial noise distribution model. .
在上述实施例中,通过样本图像训练初始噪声分布模型,并通过构建初始噪声分布模型的损失函数,以及基于构建的损失函数计算初始噪声分布模型的输出误差,基于输出误差对初始噪声分布模型进行迭代,获得符合要求的噪声分布模型。In the above embodiment, the initial noise distribution model is trained through sample images, and the loss function of the initial noise distribution model is constructed, and the output error of the initial noise distribution model is calculated based on the constructed loss function, and the initial noise distribution model is calculated based on the output error. Iterate to obtain a noise distribution model that meets the requirements.
进一步地,基于识别误差对噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型的步骤,具体包括:Further, the steps of iterating the noise distribution model based on the recognition error until the model is fitted, and obtaining the fitted noise distribution model, specifically includes:
将识别误差与预设误差阈值进行比对;Compare the recognition error with the preset error threshold;
若识别误差大于预设误差阈值,则基于反向传播算法对初始噪声分布模型进行迭代更新,直至识别误差小于或等于预设误差阈值为止,得到输出拟合的噪声分布模型;If the recognition error is greater than the preset error threshold, the initial noise distribution model is iteratively updated based on the back-propagation algorithm until the recognition error is less than or equal to the preset error threshold, and the output fitted noise distribution model is obtained;
输出噪声分布模型。Output noise distribution model.
其中,反向传播算法,即误差反向传播算法(Backpropagationalgorithm,BP算法)适合于多层神经元网络的一种学习算法,它建立在梯度下降法的基础上,用于深度学习网络的误差计算。BP网络的输入、输出关系实质上是一种映射关系:一个n输入m输出的BP神经网络所完成的功能是从n维欧氏空间向m维欧氏空间中一有限域的连续映射,这一映射具有高度非线性。BP算法的学习过程由正向传播过程和反向传播过程组成。在正向传播过程中,输入信息通过输入层经隐含层,逐层处理并传向输出层,并转入反向传播,逐层求出目标函数对各神经元权值的偏导数,构成目标函数对权值向量的梯量,以作为修改权值的依据。Among them, the backpropagation algorithm, that is, the error backpropagation algorithm (Backpropagationalgorithm, BP algorithm) is a learning algorithm suitable for multi-layer neuron networks. It is based on the gradient descent method and is used for the error calculation of deep learning networks. . The input and output relationship of BP network is essentially a mapping relationship: the function completed by a BP neural network with n input and m output is a continuous mapping from n-dimensional Euclidean space to a finite field in m-dimensional Euclidean space. A map is highly nonlinear. The learning process of BP algorithm consists of forward propagation process and back propagation process. In the process of forward propagation, the input information is processed layer by layer through the hidden layer through the input layer and transmitted to the output layer, and then transferred to the back propagation, and the partial derivative of the objective function to the weight of each neuron is obtained layer by layer, which constitutes The gradient of the objective function to the weight vector is used as the basis for modifying the weight.
具体的,将识别误差与预设误差阈值进行比较,若识别误差大于预设误差阈值,则基于反向传播算法对训练完成的初始噪声分布模型进行迭代更新,直到识别误差小于或等于预设误差阈值为止,获取输出拟合的噪声分布模型。其中,预设误差阈值可以提前设定。在上述实施例中,通过反向传播算法对训练完成的初始噪声分布模型进行验证和迭代,得到符合要求的噪声分布模型。Specifically, the identification error is compared with a preset error threshold, and if the identification error is greater than the preset error threshold, the initial noise distribution model after training is iteratively updated based on the back-propagation algorithm until the identification error is less than or equal to the preset error Up to the threshold, obtain the noise distribution model of the output fitting. The preset error threshold may be set in advance. In the above embodiment, the initial noise distribution model that has been trained is verified and iterated through the back-propagation algorithm to obtain a noise distribution model that meets the requirements.
进一步地,根据高斯-泊松联合噪声分布函数获取待降噪图像的噪声分布信息的步骤,具体包括:Further, the step of obtaining the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function specifically includes:
基于无噪声图像的像素信息对高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵;The Gauss-Poisson joint noise distribution function is randomly sampled based on the pixel information of the noise-free image, and the sampling matrix is obtained;
根据抽样矩阵得到待降噪图像的噪声分布信息。The noise distribution information of the image to be denoised is obtained according to the sampling matrix.
具体的,基于无噪声图像的像素信息对高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵,根据抽样矩阵得到待降噪图像的噪声分布信息。例如,输入的无噪声图像为512*512大小的图像,则需要对高斯-泊松联合噪声分布函数进行512*512次随机抽样,将随机抽样获得的结果进行组合,得到512*512抽样矩阵,将抽样矩阵添加到无噪声图像中,并添加与待降噪图像相关的相机噪声、格式转化噪声以及压缩噪声,最终模拟得到一 张与待降噪图像相对应的含有噪声的真实图像,基于含噪声的真实图像和无噪声图像得到待降噪图像的噪声分布信息。Specifically, the Gauss-Poisson joint noise distribution function is randomly sampled based on the pixel information of the noise-free image to obtain a sampling matrix, and the noise distribution information of the image to be denoised is obtained according to the sampling matrix. For example, if the input noise-free image is an image of size 512*512, the Gauss-Poisson joint noise distribution function needs to be randomly sampled 512*512 times, and the results obtained by random sampling are combined to obtain a 512*512 sampling matrix, The sampling matrix is added to the noise-free image, and the camera noise, format conversion noise and compression noise related to the image to be de-noised are added, and finally a real image with noise corresponding to the image to be de-noised is obtained by simulation. The noise distribution information of the image to be denoised is obtained from the noise-free real image and the noise-free image.
在上述实施例中,通过对高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵,通过抽样矩阵添加到无噪声图像中,并添加与待降噪图像相关的相机噪声、格式转化噪声以及压缩噪声,最终模拟得到一张与待降噪图像相对应的含有噪声的真实图像,基于含噪声的真实图像和无噪声图像可以获得待降噪图像的噪声分布信息。In the above embodiment, a sampling matrix is obtained by randomly sampling the Gauss-Poisson joint noise distribution function, and the sampling matrix is added to the noise-free image, and the camera noise, format conversion noise and Compress the noise, and finally simulate a real image with noise corresponding to the image to be denoised. Based on the real image with noise and the noise-free image, the noise distribution information of the image to be de-noised can be obtained.
进一步地,根据抽样矩阵得到待降噪图像的噪声分布信息的步骤,具体包括:Further, the step of obtaining the noise distribution information of the image to be denoised according to the sampling matrix specifically includes:
获取无噪声图像的图像矩阵,融合无噪声图像的图像矩阵和抽样矩阵,得到第一噪声图像;Obtain the image matrix of the noise-free image, and fuse the image matrix and the sampling matrix of the noise-free image to obtain the first noise image;
获取待降噪图像对应的相机响应函数,并基于相机响应函数对第一噪声图像处理,得到第二噪声图像;obtaining a camera response function corresponding to the image to be denoised, and processing the first noise image based on the camera response function to obtain a second noise image;
获取待降噪图像对应的格式转化信息,并基于格式转化信息对第二噪声图像进行彩色插值,得到第三噪声图像;acquiring format conversion information corresponding to the image to be denoised, and performing color interpolation on the second noise image based on the format conversion information to obtain a third noise image;
获取待降噪图像对应的压缩参数,并基于压缩参数对第三噪声图像进行压缩,得到第四噪声图像;obtaining compression parameters corresponding to the image to be denoised, and compressing the third noise image based on the compression parameters to obtain a fourth noise image;
获取第四噪声图像的的图像矩阵,并基于第四噪声图像的的图像矩阵得到待降噪图像的噪声分布信息。An image matrix of the fourth noise image is obtained, and noise distribution information of the image to be denoised is obtained based on the image matrix of the fourth noise image.
具体的,获取无噪声图像的图像矩阵,融合无噪声图像的图像矩阵和抽样矩阵,得到第一噪声图像,考虑到实际情况中相机拍摄因素、格式转化因素和图像JPEG压缩因素,需要根据上述影响因素依次对第一噪声图像进行处理,模拟一张与待降噪图像相对应的含有噪声的真实图像,其中,格式转化从原相机的bayer图像转化到到RGB图的过程,格式转化过程会产生一定的噪声,图像JPEG压缩指的是图像传输前的压缩过程,图像压缩会产生一定的噪声。在本申请具体的实施例中,根据上述影响因素依次对第一噪声图像进行处理具体运算过程如下:Specifically, the image matrix of the noise-free image is obtained, and the image matrix and the sampling matrix of the noise-free image are fused to obtain the first noise image. Considering the camera shooting factors, format conversion factors and image JPEG compression factors in the actual situation, it is necessary to take into account the above influences The factors process the first noise image in turn to simulate a real image with noise corresponding to the image to be denoised. The process of format conversion from the bayer image of the original camera to the RGB image, the format conversion process will generate Certain noise, image JPEG compression refers to the compression process before image transmission, and image compression will produce certain noise. In a specific embodiment of the present application, the specific operation process of sequentially processing the first noise image according to the above-mentioned influencing factors is as follows:
y=JPEG{f[DM(L+n(L))]}y=JPEG{f[DM(L+n(L))]}
其中,L认为是理想的无噪声图像,这里y是含噪声真实图像,f是相机响应函数,DM是指从bayer图像到RGB图像的过程,即彩色插值过程,另外考虑到实际处理的图像通常为JPEG格式,JPEG即图像压缩过程。至此,模拟图像噪声的步骤完成,得到一张与待降噪图像相对应的含有噪声的真实图像。Among them, L is considered to be an ideal noise-free image, where y is the real image with noise, f is the camera response function, and DM refers to the process from the bayer image to the RGB image, that is, the color interpolation process. For the JPEG format, JPEG is the image compression process. So far, the step of simulating image noise is completed, and a real image containing noise corresponding to the image to be denoised is obtained.
在上述实施例中,通过基于相机拍摄因素、格式转化因素和图像JPEG压缩因素对第一噪声图像进行处理,使得第一噪声图像获得相机噪声、格式转化噪声和压缩噪声,最终模拟得到一张与待降噪图像相对应的含有噪声的真实图像,基于含噪声的真实图像和无噪声图像可以获得待降噪图像的噪声分布信息。In the above embodiment, the first noise image is processed based on the camera shooting factor, the format conversion factor and the image JPEG compression factor, so that the first noise image obtains the camera noise, the format conversion noise and the compression noise, and finally obtains an image with The noise-containing real image corresponding to the image to be de-noised can be obtained based on the noise-containing real image and the noise-free image to obtain noise distribution information of the image to be de-noised.
进一步地,将待降噪图像和噪声分布信息导入到预先训练好的降噪模型,根据噪声分布信息在以降噪模型中对待降噪图像进行降噪,得到去噪图像的步骤,具体包括:Further, import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain the denoised image, specifically including:
基于噪声分布信息获取噪声分布矩阵;Obtain the noise distribution matrix based on the noise distribution information;
对待降噪图像的图像矩阵和噪声分布矩阵进行矩阵拼接,得到矩阵拼接张量;Perform matrix splicing on the image matrix and noise distribution matrix of the image to be denoised to obtain a matrix splicing tensor;
利用降噪模型的卷积核对矩阵拼接张量进行卷积运算,得到卷积运算结果;Use the convolution check of the noise reduction model to perform the convolution operation on the matrix splicing tensor to obtain the result of the convolution operation;
基于卷积运算结果进行图像重建,得到去噪图像。Image reconstruction is performed based on the result of the convolution operation to obtain a denoised image.
具体的,基于噪声分布信息获取噪声分布矩阵,对待降噪图像的图像矩阵和噪声分布矩阵进行矩阵拼接,得到矩阵拼接张量,矩阵拼接张量是一个三维的条件张量,利用降噪模型的卷积核对矩阵拼接张量进行卷积运算,得到卷积运算结果,将卷积运算结果依次填入一个空白矩阵的矩阵体中,其中,空白矩阵与待降噪图像的图像矩阵大小一致,例如都是512*512大小的矩阵,上述过程相当于对图像进行重建,重建后得到去噪图像。这里利 用条件引导的方式对矩阵拼接张量进行卷积,然后再基于卷积运算结果进行图像重建,得到去噪图像。Specifically, the noise distribution matrix is obtained based on the noise distribution information, and the image matrix of the image to be denoised and the noise distribution matrix are matrix spliced to obtain a matrix splicing tensor. The matrix splicing tensor is a three-dimensional condition tensor. The convolution kernel performs the convolution operation on the matrix splicing tensor to obtain the convolution operation result, and fills the convolution operation result into the matrix body of a blank matrix in turn, where the blank matrix is the same size as the image matrix of the image to be denoised, for example They are all 512*512-sized matrices. The above process is equivalent to reconstructing the image, and the denoised image is obtained after reconstruction. Here, the matrix splicing tensor is convolved in a condition-guided manner, and then the image is reconstructed based on the result of the convolution operation to obtain a denoised image.
在绝大多数降噪场景中,针对降噪使用的是默认为均匀噪声分布,这在自然图像一般场景中是适用的,但是在移动端场景中,由于拍摄角、光线分布等原因,会造成噪声分布并不均匀,此时就应当基于噪声分布来进行降噪,简而言之就是噪声越多的地方应当降噪力度加大,而噪声较少的地方降噪力度减小,这样才能确保获得的图像是清晰、干净,且畸变的图像。而想要基于噪声分布来进行降噪首先需要估计图像的噪声分布,然后基于噪声分布进行降噪。In most noise reduction scenes, the default is uniform noise distribution for noise reduction, which is applicable in general scenes of natural images, but in mobile scenes, due to shooting angle, light distribution and other reasons, it will cause The noise distribution is not uniform. At this time, noise reduction should be carried out based on the noise distribution. In short, the noise reduction should be increased in places with more noise, and the noise reduction should be reduced in places with less noise. The obtained images are sharp, clean, and distorted images. To perform noise reduction based on the noise distribution, it is necessary to first estimate the noise distribution of the image, and then perform noise reduction based on the noise distribution.
针对上述技术问题,本申请公开了一种基于噪声场的图像降噪方法、装置、设备及存储介质,属于人工智能技术领域,本申请通过将与待降噪图像内容相近或者相同的无噪声图像输入到预设的噪声分布模型得到一个高斯-泊松联合噪声分布函数,通过对高斯-泊松联合噪声分布函数进行随机抽样,模拟得到待降噪图像的噪声分布信息,然后预先训练的降噪模型根据噪声分布信息对待降噪图像进行降噪,得到去噪图像。相比于现有的对图像整体统一进行降噪方案,本申请基于噪声分布来进行降噪,即针对噪声越多的地方加大降噪力度,而针对噪声较少的地方减小降噪力度,这样不仅能够获得清晰、干净的降噪图像,而且能够防止图像降噪过程中产生的畸变。In view of the above technical problems, the present application discloses an image noise reduction method, device, equipment and storage medium based on noise field, which belong to the technical field of artificial intelligence. Input the preset noise distribution model to obtain a Gauss-Poisson joint noise distribution function. By randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained denoising The model denoises the denoised image according to the noise distribution information, and obtains the denoised image. Compared with the existing unified denoising scheme for the whole image, the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
需要强调的是,为进一步保证上述待降噪图像和无噪声图像的私密和安全性,上述待降噪图像和无噪声图像还可以存储于一区块链的节点中。It should be emphasized that, in order to further ensure the privacy and security of the above-mentioned images to be denoised and noise-free images, the above-mentioned images to be de-noised and noise-free images can also be stored in a node of a blockchain.
本申请所指区块链是分布式数据存储、点对点传输、共识机制、加密算法等计算机技术的新型应用模式。区块链(Blockchain),本质上是一个去中心化的数据库,是一串使用密码学方法相关联产生的数据块,每一个数据块中包含了一批次网络交易的信息,用于验证其信息的有效性(防伪)和生成下一个区块。区块链可以包括区块链底层平台、平台产品服务层以及应用服务层等。The blockchain referred to in this application is a new application mode of computer technologies such as distributed data storage, point-to-point transmission, consensus mechanism, and encryption algorithm. Blockchain, essentially a decentralized database, is a series of data blocks associated with cryptographic methods. Each data block contains a batch of network transaction information to verify its Validity of information (anti-counterfeiting) and generation of the next block. The blockchain can include the underlying platform of the blockchain, the platform product service layer, and the application service layer.
本领域普通技术人员可以理解实现上述实施例方法中的全部或部分流程,是可以通过计算机可读指令来指令相关的硬件来完成,该计算机可读指令可存储于一计算机可读取存储介质中,该计算机可读指令在执行时,可包括如上述各方法的实施例的流程。其中,前述的存储介质可为磁碟、光盘、只读存储记忆体(Read-Only Memory,ROM)等非易失性存储介质,或随机存储记忆体(Random Access Memory,RAM)等。Those of ordinary skill in the art can understand that all or part of the processes in the methods of the above embodiments can be implemented by instructing relevant hardware through computer-readable instructions, and the computer-readable instructions can be stored in a computer-readable storage medium. , when the computer-readable instructions are executed, the processes of the above-mentioned method embodiments may be included. Wherein, the aforementioned storage medium may be a non-volatile storage medium such as a magnetic disk, an optical disk, a read-only memory (Read-Only Memory, ROM), or a random access memory (Random Access Memory, RAM) or the like.
应该理解的是,虽然附图的流程图中的各个步骤按照箭头的指示依次显示,但是这些步骤并不是必然按照箭头指示的顺序依次执行。除非本文中有明确的说明,这些步骤的执行并没有严格的顺序限制,其可以以其他的顺序执行。而且,附图的流程图中的至少一部分步骤可以包括多个子步骤或者多个阶段,这些子步骤或者阶段并不必然是在同一时刻执行完成,而是可以在不同的时刻执行,其执行顺序也不必然是依次进行,而是可以与其他步骤或者其他步骤的子步骤或者阶段的至少一部分轮流或者交替地执行。It should be understood that although the various steps in the flowchart of the accompanying drawings are sequentially shown in the order indicated by the arrows, these steps are not necessarily executed in sequence in the order indicated by the arrows. Unless explicitly stated herein, the execution of these steps is not strictly limited to the order and may be performed in other orders. Moreover, at least a part of the steps in the flowchart of the accompanying drawings may include multiple sub-steps or multiple stages, and these sub-steps or stages are not necessarily executed at the same time, but may be executed at different times, and the execution sequence is also It does not have to be performed sequentially, but may be performed alternately or alternately with other steps or at least a portion of sub-steps or stages of other steps.
进一步参考图3,作为对上述图2所示方法的实现,本申请提供了一种基于噪声场的图像降噪装置的一个实施例,该装置实施例与图2所示的方法实施例相对应,该装置具体可以应用于各种电子设备中。Further referring to FIG. 3 , as an implementation of the method shown in FIG. 2 above, the present application provides an embodiment of an image noise reduction device based on a noise field, and the device embodiment corresponds to the method embodiment shown in FIG. 2 . , the device can be specifically applied to various electronic devices.
如图3所示,本实施例所述的基于噪声场的图像降噪装置包括:As shown in FIG. 3 , the noise field-based image noise reduction device in this embodiment includes:
图像获取模块301,用于获取待降噪图像以及与待降噪图像相对应的无噪声图像;An image acquisition module 301, configured to acquire an image to be denoised and a noise-free image corresponding to the image to be denoised;
函数构建模块302,用于将无噪声图像输入预设的噪声分布模型,并在噪声分布模型中构建无噪声图像的高斯-泊松联合噪声分布函数;A function construction module 302, configured to input the noise-free image into a preset noise distribution model, and construct a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
噪声模拟模块303,用于根据高斯-泊松联合噪声分布函数获取待降噪图像的噪声分布信息;A noise simulation module 303, configured to obtain the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
图像降噪模块304,用于将待降噪图像和噪声分布信息导入到预先训练好的降噪模型,根据噪声分布信息在以降噪模型中对待降噪图像进行降噪,得到去噪图像。The image denoising module 304 is used to import the image to be denoised and the noise distribution information into the pre-trained denoising model, and denoise the denoising image in the denoising model according to the noise distribution information to obtain a denoised image.
进一步地,噪声分布模型中预设有一个高斯核和一个泊松核,函数构建模块302具体包括:Further, a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, and the function building module 302 specifically includes:
信息提取单元,用于将无噪声图像输入到噪声分布模型中,通过噪声分布模型获取无噪声图像的像素信息;an information extraction unit, used for inputting the noise-free image into the noise distribution model, and obtaining pixel information of the noise-free image through the noise distribution model;
函数构建单元,用于基于无噪声图像的像素信息、泊松核以及高斯核构建无噪声图像的高斯-泊松联合噪声分布函数。The function construction unit is used to construct the Gauss-Poisson joint noise distribution function of the noise-free image based on the pixel information of the noise-free image, the Poisson kernel and the Gaussian kernel.
进一步地,该基于噪声场的图像降噪装置还包括:Further, the image noise reduction device based on noise field also includes:
样本获取模块,用于从预设的图像数据库中获取样本图像,将样本图像输入到预设的初始噪声分布模型,获取初始噪声分布模型的输出结果;a sample acquisition module, configured to acquire a sample image from a preset image database, input the sample image into a preset initial noise distribution model, and obtain an output result of the initial noise distribution model;
误差计算模块,用于构建初始噪声分布模型的损失函数,基于输出结果和预设标准结果,使用噪声分布模型的损失函数进行误差计算,获取识别误差;The error calculation module is used to construct the loss function of the initial noise distribution model, and based on the output result and the preset standard result, use the loss function of the noise distribution model to perform error calculation to obtain the identification error;
模型迭代模块,用于基于识别误差对噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型。The model iteration module is used to iterate the noise distribution model based on the recognition error until the model is fitted, and obtain the output fitted noise distribution model.
进一步地,模型迭代模块具体包括:Further, the model iteration module specifically includes:
误差比对单元,用于将识别误差与预设误差阈值进行比对;The error comparison unit is used to compare the recognition error with the preset error threshold;
模型迭代单元,用于当识别误差大于预设误差阈值时,基于反向传播算法对初始噪声分布模型进行迭代更新,直至识别误差小于或等于预设误差阈值为止,得到输出拟合的噪声分布模型;The model iteration unit is used to iteratively update the initial noise distribution model based on the back-propagation algorithm when the recognition error is greater than the preset error threshold, until the recognition error is less than or equal to the preset error threshold, and obtain the output fitted noise distribution model ;
模型输出单元,用于输出噪声分布模型。Model output unit for outputting the noise distribution model.
进一步地,噪声模拟模块303具体包括:Further, the noise simulation module 303 specifically includes:
随机抽样单元,用于基于无噪声图像的像素信息对高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵;The random sampling unit is used to randomly sample the Gauss-Poisson joint noise distribution function based on the pixel information of the noise-free image to obtain a sampling matrix;
噪声模拟单元,用于根据抽样矩阵得到待降噪图像的噪声分布信息。The noise simulation unit is used to obtain the noise distribution information of the image to be denoised according to the sampling matrix.
进一步地,噪声模拟单元具体包括:Further, the noise simulation unit specifically includes:
矩阵融合子单元,用于获取无噪声图像的图像矩阵,融合无噪声图像的图像矩阵和抽样矩阵,得到第一噪声图像;The matrix fusion subunit is used to obtain the image matrix of the noise-free image, and fuse the image matrix and the sampling matrix of the noise-free image to obtain the first noise image;
相机噪声模拟子单元,用于获取待降噪图像对应的相机响应函数,并基于相机响应函数对第一噪声图像处理,得到第二噪声图像;a camera noise simulation subunit, configured to obtain a camera response function corresponding to the image to be denoised, and process the first noise image based on the camera response function to obtain a second noise image;
格式转化噪声模拟子单元,用于获取待降噪图像对应的格式转化信息,并基于格式转化信息对第二噪声图像进行彩色插值,得到第三噪声图像;a format conversion noise simulation subunit, used for acquiring format conversion information corresponding to the image to be denoised, and performing color interpolation on the second noise image based on the format conversion information to obtain a third noise image;
压缩噪声模拟子单元,用于获取待降噪图像对应的压缩参数,并基于压缩参数对第三噪声图像进行压缩,得到第四噪声图像;A compression noise simulation subunit, used for acquiring compression parameters corresponding to the image to be denoised, and compressing the third noise image based on the compression parameters to obtain a fourth noise image;
噪声分布子单元,用于获取第四噪声图像的的图像矩阵,并基于第四噪声图像的的图像矩阵得到待降噪图像的噪声分布信息。The noise distribution subunit is configured to acquire an image matrix of the fourth noise image, and obtain noise distribution information of the image to be denoised based on the image matrix of the fourth noise image.
进一步地,图像降噪模块304具体包括:Further, the image noise reduction module 304 specifically includes:
分布矩阵单元,用于基于噪声分布信息获取噪声分布矩阵;a distribution matrix unit, used to obtain a noise distribution matrix based on the noise distribution information;
矩阵拼接单元,用于对待降噪图像的图像矩阵和噪声分布矩阵进行矩阵拼接,得到矩阵拼接张量;The matrix splicing unit is used to perform matrix splicing of the image matrix and the noise distribution matrix of the image to be denoised to obtain a matrix splicing tensor;
卷积运算单元,用于利用降噪模型的卷积核对矩阵拼接张量进行卷积运算,得到卷积运算结果;The convolution operation unit is used to perform the convolution operation on the matrix splicing tensor by using the convolution check of the noise reduction model to obtain the result of the convolution operation;
图像重建单元,用于基于卷积运算结果进行图像重建,得到去噪图像。The image reconstruction unit is used for image reconstruction based on the result of the convolution operation to obtain a denoised image.
本申请公开了一种基于噪声场的图像降噪装置,属于人工智能技术领域,本申请通过将与待降噪图像内容相近或者相同的无噪声图像输入到预设的噪声分布模型得到一个高斯-泊松联合噪声分布函数,通过对高斯-泊松联合噪声分布函数进行随机抽样,模拟得到待降噪图像的噪声分布信息,然后预先训练的降噪模型根据噪声分布信息对待降噪图像进行降噪,得到去噪图像。相比于现有的对图像整体统一进行降噪方案,本申请基于噪声分 布来进行降噪,即针对噪声越多的地方加大降噪力度,而针对噪声较少的地方减小降噪力度,这样不仅能够获得清晰、干净的降噪图像,而且能够防止图像降噪过程中产生的畸变。The present application discloses an image noise reduction device based on noise field, which belongs to the technical field of artificial intelligence. The present application obtains a Gauss- Poisson joint noise distribution function, by randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model denoises the image to be denoised according to the noise distribution information. , to get the denoised image. Compared with the existing unified denoising scheme for the whole image, the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
为解决上述技术问题,本申请实施例还提供计算机设备。具体请参阅图4,图4为本实施例计算机设备基本结构框图。To solve the above technical problems, the embodiments of the present application also provide computer equipment. Please refer to FIG. 4 for details. FIG. 4 is a block diagram of a basic structure of a computer device according to this embodiment.
所述计算机设备4包括通过系统总线相互通信连接存储器41、处理器42、网络接口43。需要指出的是,图中仅示出了具有组件41-43的计算机设备4,但是应理解的是,并不要求实施所有示出的组件,可以替代的实施更多或者更少的组件。其中,本技术领域技术人员可以理解,这里的计算机设备是一种能够按照事先设定或存储的指令,自动进行数值计算和/或信息处理的设备,其硬件包括但不限于微处理器、专用集成电路(Application Specific Integrated Circuit,ASIC)、可编程门阵列(Field-Programmable Gate Array,FPGA)、数字处理器(Digital Signal Processor,DSP)、嵌入式设备等。The computer device 4 includes a memory 41, a processor 42, and a network interface 43 that communicate with each other through a system bus. It should be noted that only the computer device 4 with components 41-43 is shown in the figure, but it should be understood that it is not required to implement all of the shown components, and more or less components may be implemented instead. Among them, those skilled in the art can understand that the computer device here is a device that can automatically perform numerical calculation and/or information processing according to pre-set or stored instructions, and its hardware includes but is not limited to microprocessors, special-purpose Integrated circuit (Application Specific Integrated Circuit, ASIC), programmable gate array (Field-Programmable Gate Array, FPGA), digital processor (Digital Signal Processor, DSP), embedded equipment, etc.
所述计算机设备可以是桌上型计算机、笔记本、掌上电脑及云端服务器等计算设备。所述计算机设备可以与用户通过键盘、鼠标、遥控器、触摸板或声控设备等方式进行人机交互。The computer equipment may be a desktop computer, a notebook computer, a palmtop computer, a cloud server and other computing equipment. The computer device can perform human-computer interaction with the user through a keyboard, a mouse, a remote control, a touch pad or a voice control device.
所述存储器41至少包括一种类型的可读存储介质,所述可读存储介质包括闪存、硬盘、多媒体卡、卡型存储器(例如,SD或DX存储器等)、随机访问存储器(RAM)、静态随机访问存储器(SRAM)、只读存储器(ROM)、电可擦除可编程只读存储器(EEPROM)、可编程只读存储器(PROM)、磁性存储器、磁盘、光盘等。在一些实施例中,所述存储器41可以是所述计算机设备4的内部存储单元,例如该计算机设备4的硬盘或内存。在另一些实施例中,所述存储器41也可以是所述计算机设备4的外部存储设备,例如该计算机设备4上配备的插接式硬盘,智能存储卡(Smart Media Card,SMC),安全数字(Secure Digital,SD)卡,闪存卡(Flash Card)等。当然,所述存储器41还可以既包括所述计算机设备4的内部存储单元也包括其外部存储设备。本实施例中,所述存储器41通常用于存储安装于所述计算机设备4的操作系统和各类应用软件,例如基于噪声场的图像降噪方法的计算机可读指令等。此外,所述存储器41还可以用于暂时地存储已经输出或者将要输出的各类数据。The memory 41 includes at least one type of readable storage medium, and the readable storage medium includes flash memory, hard disk, multimedia card, card-type memory (for example, SD or DX memory, etc.), random access memory (RAM), static Random Access Memory (SRAM), Read Only Memory (ROM), Electrically Erasable Programmable Read Only Memory (EEPROM), Programmable Read Only Memory (PROM), Magnetic Memory, Magnetic Disk, Optical Disk, etc. In some embodiments, the memory 41 may be an internal storage unit of the computer device 4 , such as a hard disk or a memory of the computer device 4 . In other embodiments, the memory 41 may also be an external storage device of the computer device 4, such as a plug-in hard disk, a smart memory card (Smart Media Card, SMC), a secure digital (Secure Digital, SD) card, flash memory card (Flash Card), etc. Of course, the memory 41 may also include both the internal storage unit of the computer device 4 and its external storage device. In this embodiment, the memory 41 is generally used to store the operating system and various application software installed on the computer device 4 , such as computer-readable instructions for an image noise reduction method based on a noise field. In addition, the memory 41 can also be used to temporarily store various types of data that have been output or will be output.
所述处理器42在一些实施例中可以是中央处理器(Central Processing Unit,CPU)、控制器、微控制器、微处理器、或其他数据处理芯片。该处理器42通常用于控制所述计算机设备4的总体操作。本实施例中,所述处理器42用于运行所述存储器41中存储的计算机可读指令或者处理数据,例如运行所述基于噪声场的图像降噪方法的计算机可读指令。The processor 42 may be a central processing unit (Central Processing Unit, CPU), a controller, a microcontroller, a microprocessor, or other data processing chips in some embodiments. This processor 42 is typically used to control the overall operation of the computer device 4 . In this embodiment, the processor 42 is configured to execute computer-readable instructions stored in the memory 41 or process data, for example, computer-readable instructions for executing the noise field-based image noise reduction method.
所述网络接口43可包括无线网络接口或有线网络接口,该网络接口43通常用于在所述计算机设备4与其他电子设备之间建立通信连接。The network interface 43 may include a wireless network interface or a wired network interface, and the network interface 43 is generally used to establish a communication connection between the computer device 4 and other electronic devices.
本申请公开了一种计算机设备,属于人工智能技术领域,本申请通过将与待降噪图像内容相近或者相同的无噪声图像输入到预设的噪声分布模型得到一个高斯-泊松联合噪声分布函数,通过对高斯-泊松联合噪声分布函数进行随机抽样,模拟得到待降噪图像的噪声分布信息,然后预先训练的降噪模型根据噪声分布信息对待降噪图像进行降噪,得到去噪图像。相比于现有的对图像整体统一进行降噪方案,本申请基于噪声分布来进行降噪,即针对噪声越多的地方加大降噪力度,而针对噪声较少的地方减小降噪力度,这样不仅能够获得清晰、干净的降噪图像,而且能够防止图像降噪过程中产生的畸变。The present application discloses a computer device, which belongs to the technical field of artificial intelligence. The present application obtains a Gauss-Poisson joint noise distribution function by inputting a noise-free image with similar or identical content to the image to be denoised into a preset noise distribution model , by randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model denoises the denoised image according to the noise distribution information to obtain a denoised image. Compared with the existing unified denoising scheme for the whole image, the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
本申请还提供了另一种实施方式,即提供一种计算机可读存储介质,所述计算机可读存储介质可以是非易失性,也可以是易失性,所述计算机可读存储介质存储有计算机可读指令,所述计算机可读指令可被至少一个处理器执行,以使所述至少一个处理器执行如上述的基于噪声场的图像降噪方法的步骤。The present application also provides another implementation manner, that is, to provide a computer-readable storage medium, the computer-readable storage medium may be non-volatile or volatile, and the computer-readable storage medium stores Computer readable instructions executable by at least one processor to cause the at least one processor to perform the steps of the noise field based image noise reduction method as described above.
本申请公开了一种存储介质,属于人工智能技术领域,本申请通过将与待降噪图像内容相近或者相同的无噪声图像输入到预设的噪声分布模型得到一个高斯-泊松联合噪声分 布函数,通过对高斯-泊松联合噪声分布函数进行随机抽样,模拟得到待降噪图像的噪声分布信息,然后预先训练的降噪模型根据噪声分布信息对待降噪图像进行降噪,得到去噪图像。相比于现有的对图像整体统一进行降噪方案,本申请基于噪声分布来进行降噪,即针对噪声越多的地方加大降噪力度,而针对噪声较少的地方减小降噪力度,这样不仅能够获得清晰、干净的降噪图像,而且能够防止图像降噪过程中产生的畸变。The present application discloses a storage medium, which belongs to the technical field of artificial intelligence. The present application obtains a Gauss-Poisson joint noise distribution function by inputting a noise-free image that is similar or identical to the content of the image to be de-noised into a preset noise distribution model , by randomly sampling the Gauss-Poisson joint noise distribution function, the noise distribution information of the image to be denoised is obtained by simulation, and then the pre-trained noise reduction model denoises the denoised image according to the noise distribution information to obtain a denoised image. Compared with the existing unified denoising scheme for the whole image, the present application performs denoising based on the noise distribution, that is, the noise reduction intensity is increased for the places with more noise, and the noise reduction intensity is reduced for the places with less noise. , which can not only obtain a clear and clean noise-reduced image, but also prevent the distortion generated in the process of image noise reduction.
通过以上的实施方式的描述,本领域的技术人员可以清楚地了解到上述实施例方法可借助软件加必需的通用硬件平台的方式来实现,当然也可以通过硬件,但很多情况下前者是更佳的实施方式。基于这样的理解,本申请的技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质(如ROM/RAM、磁碟、光盘)中,包括若干指令用以使得一台终端设备(可以是手机,计算机,服务器,空调器,或者网络设备等)执行本申请各个实施例所述的方法。From the description of the above embodiments, those skilled in the art can clearly understand that the method of the above embodiment can be implemented by means of software plus a necessary general hardware platform, and of course can also be implemented by hardware, but in many cases the former is better implementation. Based on this understanding, the technical solution of the present application can be embodied in the form of a software product in essence or in a part that contributes to the prior art, and the computer software product is stored in a storage medium (such as ROM/RAM, magnetic disk, CD-ROM), including several instructions to make a terminal device (which may be a mobile phone, a computer, a server, an air conditioner, or a network device, etc.) execute the methods described in the various embodiments of this application.
显然,以上所描述的实施例仅仅是本申请一部分实施例,而不是全部的实施例,附图中给出了本申请的较佳实施例,但并不限制本申请的专利范围。本申请可以以许多不同的形式来实现,相反地,提供这些实施例的目的是使对本申请的公开内容的理解更加透彻全面。尽管参照前述实施例对本申请进行了详细的说明,对于本领域的技术人员来而言,其依然可以对前述各具体实施方式所记载的技术方案进行修改,或者对其中部分技术特征进行等效替换。凡是利用本申请说明书及附图内容所做的等效结构,直接或间接运用在其他相关的技术领域,均同理在本申请专利保护范围之内。Obviously, the above-described embodiments are only a part of the embodiments of the present application, rather than all of the embodiments. The accompanying drawings show the preferred embodiments of the present application, but do not limit the scope of the patent of the present application. This application may be embodied in many different forms, rather these embodiments are provided so that a thorough and complete understanding of the disclosure of this application is provided. Although the present application has been described in detail with reference to the foregoing embodiments, those skilled in the art can still modify the technical solutions described in the foregoing specific embodiments, or perform equivalent replacements for some of the technical features. . Any equivalent structure made by using the contents of the description and drawings of the present application, which is directly or indirectly used in other related technical fields, is also within the scope of protection of the patent of the present application.

Claims (20)

  1. 一种基于噪声场的图像降噪方法,包括:An image noise reduction method based on noise field, comprising:
    获取待降噪图像以及与所述待降噪图像相对应的无噪声图像;acquiring an image to be denoised and a noise-free image corresponding to the image to be denoised;
    将所述无噪声图像输入预设的噪声分布模型,并在所述噪声分布模型中构建所述无噪声图像的高斯-泊松联合噪声分布函数;Inputting the noise-free image into a preset noise distribution model, and constructing a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
    根据所述高斯-泊松联合噪声分布函数获取所述待降噪图像的噪声分布信息;Acquiring noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
    将所述待降噪图像和所述噪声分布信息导入到预先训练好的降噪模型,根据所述噪声分布信息在以所述降噪模型中对所述待降噪图像进行降噪,得到去噪图像。Import the image to be denoised and the noise distribution information into a pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain a denoised image. noisy image.
  2. 如权利要求1所述的基于噪声场的图像降噪方法,其中,所述噪声分布模型中预设有一个高斯核和一个泊松核,所述将所述无噪声图像输入预设的噪声分布模型,并在所述噪声分布模型中构建所述无噪声图像的高斯-泊松联合噪声分布函数的步骤,具体包括:The noise field-based image denoising method according to claim 1, wherein a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, and the noise-free image is input into a preset noise distribution The steps of constructing the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model include:
    将所述无噪声图像输入到所述噪声分布模型中,通过所述噪声分布模型获取所述无噪声图像的像素信息;inputting the noise-free image into the noise distribution model, and obtaining pixel information of the noise-free image through the noise distribution model;
    基于所述无噪声图像的像素信息、所述泊松核以及所述高斯核构建所述无噪声图像的高斯-泊松联合噪声分布函数。A Gauss-Poisson joint noise distribution function of the noise-free image is constructed based on the pixel information of the noise-free image, the Poisson kernel, and the Gaussian kernel.
  3. 如权利要求2所述的基于噪声场的图像降噪方法,其中,在所述将所述无噪声图像输入到所述噪声分布模型中,通过所述噪声分布模型获取所述无噪声图像的像素信息的步骤之前,还包括:The noise-field-based image denoising method according to claim 2, wherein, when the noise-free image is input into the noise distribution model, pixels of the noise-free image are obtained through the noise distribution model Before the steps of the information, also include:
    从预设的图像数据库中获取样本图像,将所述样本图像输入到预设的初始噪声分布模型,获取所述初始噪声分布模型的输出结果;Obtain a sample image from a preset image database, input the sample image into a preset initial noise distribution model, and obtain an output result of the initial noise distribution model;
    构建所述初始噪声分布模型的损失函数,基于所述输出结果和预设标准结果,使用所述噪声分布模型的损失函数进行误差计算,获取识别误差;constructing a loss function of the initial noise distribution model, and using the loss function of the noise distribution model to perform error calculation based on the output result and the preset standard result to obtain an identification error;
    基于所述识别误差对所述噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型。The noise distribution model is iterated based on the identification error until the model is fitted, and the fitted noise distribution model is output.
  4. 如权利要求1所述的基于噪声场的图像降噪方法,其中,所述基于所述识别误差对所述噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型的步骤,具体包括:The noise-field-based image denoising method according to claim 1, wherein the step of iterating the noise distribution model based on the identification error until the model is fitted to obtain a fitted noise distribution model, Specifically include:
    将所述识别误差与预设误差阈值进行比对;comparing the identification error with a preset error threshold;
    若所述识别误差大于预设误差阈值,则基于反向传播算法对所述初始噪声分布模型进行迭代更新,直至所述识别误差小于或等于预设误差阈值为止,得到输出拟合的噪声分布模型;If the identification error is greater than a preset error threshold, the initial noise distribution model is iteratively updated based on the back-propagation algorithm until the identification error is less than or equal to the preset error threshold, and an output fitted noise distribution model is obtained ;
    输出所述噪声分布模型。The noise distribution model is output.
  5. 如权利要求1至4任意一项所述的基于噪声场的图像降噪方法,其中,所述根据所述高斯-泊松联合噪声分布函数获取所述待降噪图像的噪声分布信息的步骤,具体包括:The noise field-based image denoising method according to any one of claims 1 to 4, wherein the step of acquiring the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function, Specifically include:
    基于所述无噪声图像的像素信息对所述高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵;Randomly sample the Gauss-Poisson joint noise distribution function based on the pixel information of the noise-free image to obtain a sampling matrix;
    根据所述抽样矩阵得到所述待降噪图像的噪声分布信息。The noise distribution information of the image to be denoised is obtained according to the sampling matrix.
  6. 如权利要求5所述的基于噪声场的图像降噪方法,其中,所述根据所述抽样矩阵得到所述待降噪图像的噪声分布信息的步骤,具体包括:The noise field-based image denoising method according to claim 5, wherein the step of obtaining the noise distribution information of the image to be denoised according to the sampling matrix specifically includes:
    获取所述无噪声图像的图像矩阵,融合所述无噪声图像的图像矩阵和所述抽样矩阵,得到第一噪声图像;obtaining the image matrix of the noise-free image, and fusing the image matrix of the noise-free image and the sampling matrix to obtain a first noise image;
    获取所述待降噪图像对应的相机响应函数,并基于所述相机响应函数对所述第一噪声图像处理,得到第二噪声图像;acquiring a camera response function corresponding to the image to be denoised, and processing the first noise image based on the camera response function to obtain a second noise image;
    获取所述待降噪图像对应的格式转化信息,并基于所述格式转化信息对所述第二噪声图像进行彩色插值,得到第三噪声图像;acquiring format conversion information corresponding to the image to be denoised, and performing color interpolation on the second noise image based on the format conversion information to obtain a third noise image;
    获取所述待降噪图像对应的压缩参数,并基于所述压缩参数对所述第三噪声图像进行压缩,得到第四噪声图像;Acquiring compression parameters corresponding to the image to be denoised, and compressing the third noise image based on the compression parameters to obtain a fourth noise image;
    获取所述第四噪声图像的的图像矩阵,并基于所述第四噪声图像的的图像矩阵得到所述待降噪图像的噪声分布信息。An image matrix of the fourth noise image is acquired, and noise distribution information of the image to be denoised is obtained based on the image matrix of the fourth noise image.
  7. 如权利要求5所述的基于噪声场的图像降噪方法,其中,所述将所述待降噪图像和所述噪声分布信息导入到预先训练好的降噪模型,根据所述噪声分布信息在以所述降噪模型中对所述待降噪图像进行降噪,得到去噪图像的步骤,具体包括:The image noise reduction method based on the noise field according to claim 5, wherein the image to be denoised and the noise distribution information are imported into a pre-trained noise reduction model, and the noise distribution information is based on the noise distribution information. The steps of denoising the image to be denoised in the denoising model to obtain the denoising image specifically include:
    基于所述噪声分布信息获取噪声分布矩阵;obtaining a noise distribution matrix based on the noise distribution information;
    对所述待降噪图像的图像矩阵和所述噪声分布矩阵进行矩阵拼接,得到矩阵拼接张量;performing matrix splicing on the image matrix of the image to be denoised and the noise distribution matrix to obtain a matrix splicing tensor;
    利用所述降噪模型的卷积核对所述矩阵拼接张量进行卷积运算,得到卷积运算结果;Use the convolution kernel of the noise reduction model to perform a convolution operation on the matrix splicing tensor to obtain a convolution operation result;
    基于所述卷积运算结果进行图像重建,得到去噪图像。Image reconstruction is performed based on the result of the convolution operation to obtain a denoised image.
  8. 一种基于噪声场的图像降噪装置,包括:An image noise reduction device based on noise field, comprising:
    图像获取模块,用于获取待降噪图像以及与所述待降噪图像相对应的无噪声图像;an image acquisition module, configured to acquire an image to be denoised and a noise-free image corresponding to the image to be denoised;
    函数构建模块,用于将所述无噪声图像输入预设的噪声分布模型,并在所述噪声分布模型中构建所述无噪声图像的高斯-泊松联合噪声分布函数;a function building module for inputting the noise-free image into a preset noise distribution model, and constructing a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
    噪声模拟模块,用于根据所述高斯-泊松联合噪声分布函数获取所述待降噪图像的噪声分布信息;a noise simulation module, configured to obtain the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
    图像降噪模块,用于将所述待降噪图像和所述噪声分布信息导入到预先训练好的降噪模型,根据所述噪声分布信息在以所述降噪模型中对所述待降噪图像进行降噪,得到去噪图像。An image noise reduction module, configured to import the image to be denoised and the noise distribution information into a pre-trained noise reduction model, and use the noise reduction model to denoise the noise reduction model according to the noise distribution information The image is denoised to obtain a denoised image.
  9. 一种计算机设备,包括存储器和处理器,所述存储器中存储有计算机可读指令,所述处理器执行所述计算机可读指令时实现如下所述的基于噪声场的图像降噪方法:A computer device, comprising a memory and a processor, wherein computer-readable instructions are stored in the memory, and the processor implements the following noise field-based image noise reduction method when executing the computer-readable instructions:
    获取待降噪图像以及与所述待降噪图像相对应的无噪声图像;acquiring an image to be denoised and a noise-free image corresponding to the image to be denoised;
    将所述无噪声图像输入预设的噪声分布模型,并在所述噪声分布模型中构建所述无噪声图像的高斯-泊松联合噪声分布函数;Inputting the noise-free image into a preset noise distribution model, and constructing a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
    根据所述高斯-泊松联合噪声分布函数获取所述待降噪图像的噪声分布信息;Acquiring noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
    将所述待降噪图像和所述噪声分布信息导入到预先训练好的降噪模型,根据所述噪声分布信息在以所述降噪模型中对所述待降噪图像进行降噪,得到去噪图像。Import the image to be denoised and the noise distribution information into a pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain a denoised image. noisy image.
  10. 如权利要求9所述的计算机设备,其中,所述噪声分布模型中预设有一个高斯核和一个泊松核,所述将所述无噪声图像输入预设的噪声分布模型,并在所述噪声分布模型中构建所述无噪声图像的高斯-泊松联合噪声分布函数的步骤,具体包括:The computer device according to claim 9, wherein a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, the noise-free image is input into the preset noise distribution model, and the noise-free image is input in the preset noise distribution model. The steps of constructing the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model specifically include:
    将所述无噪声图像输入到所述噪声分布模型中,通过所述噪声分布模型获取所述无噪声图像的像素信息;inputting the noise-free image into the noise distribution model, and obtaining pixel information of the noise-free image through the noise distribution model;
    基于所述无噪声图像的像素信息、所述泊松核以及所述高斯核构建所述无噪声图像的高斯-泊松联合噪声分布函数。A Gauss-Poisson joint noise distribution function of the noise-free image is constructed based on the pixel information of the noise-free image, the Poisson kernel, and the Gaussian kernel.
  11. 如权利要求10所述的计算机设备,其中,在所述将所述无噪声图像输入到所述噪声分布模型中,通过所述噪声分布模型获取所述无噪声图像的像素信息的步骤之前,还包括:The computer device of claim 10, wherein before the step of inputting the noise-free image into the noise distribution model, and obtaining pixel information of the noise-free image through the noise distribution model, further include:
    从预设的图像数据库中获取样本图像,将所述样本图像输入到预设的初始噪声分布模型,获取所述初始噪声分布模型的输出结果;Obtain a sample image from a preset image database, input the sample image into a preset initial noise distribution model, and obtain an output result of the initial noise distribution model;
    构建所述初始噪声分布模型的损失函数,基于所述输出结果和预设标准结果,使用所述噪声分布模型的损失函数进行误差计算,获取识别误差;constructing a loss function of the initial noise distribution model, and using the loss function of the noise distribution model to perform error calculation based on the output result and the preset standard result to obtain an identification error;
    基于所述识别误差对所述噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型。The noise distribution model is iterated based on the identification error until the model is fitted, and the fitted noise distribution model is output.
  12. 如权利要求9所述的计算机设备,其中,所述基于所述识别误差对所述噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型的步骤,具体包括:The computer device according to claim 9, wherein the step of iterating the noise distribution model based on the identification error until the model is fitted to obtain the output fitted noise distribution model specifically includes:
    将所述识别误差与预设误差阈值进行比对;comparing the identification error with a preset error threshold;
    若所述识别误差大于预设误差阈值,则基于反向传播算法对所述初始噪声分布模型进行迭代更新,直至所述识别误差小于或等于预设误差阈值为止,得到输出拟合的噪声分布模型;If the identification error is greater than a preset error threshold, the initial noise distribution model is iteratively updated based on the back-propagation algorithm until the identification error is less than or equal to the preset error threshold, and an output fitted noise distribution model is obtained ;
    输出所述噪声分布模型。The noise distribution model is output.
  13. 如权利要求9至12任意一项所述的计算机设备,其中,所述根据所述高斯-泊松联合噪声分布函数获取所述待降噪图像的噪声分布信息的步骤,具体包括:The computer device according to any one of claims 9 to 12, wherein the step of acquiring the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function specifically includes:
    基于所述无噪声图像的像素信息对所述高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵;Randomly sample the Gauss-Poisson joint noise distribution function based on the pixel information of the noise-free image to obtain a sampling matrix;
    根据所述抽样矩阵得到所述待降噪图像的噪声分布信息。The noise distribution information of the image to be denoised is obtained according to the sampling matrix.
  14. 如权利要求13所述的计算机设备,其中,所述根据所述抽样矩阵得到所述待降噪图像的噪声分布信息的步骤,具体包括:The computer device according to claim 13, wherein the step of obtaining the noise distribution information of the image to be denoised according to the sampling matrix specifically comprises:
    获取所述无噪声图像的图像矩阵,融合所述无噪声图像的图像矩阵和所述抽样矩阵,得到第一噪声图像;obtaining the image matrix of the noise-free image, and fusing the image matrix of the noise-free image and the sampling matrix to obtain a first noise image;
    获取所述待降噪图像对应的相机响应函数,并基于所述相机响应函数对所述第一噪声图像处理,得到第二噪声图像;acquiring a camera response function corresponding to the image to be denoised, and processing the first noise image based on the camera response function to obtain a second noise image;
    获取所述待降噪图像对应的格式转化信息,并基于所述格式转化信息对所述第二噪声图像进行彩色插值,得到第三噪声图像;acquiring format conversion information corresponding to the image to be denoised, and performing color interpolation on the second noise image based on the format conversion information to obtain a third noise image;
    获取所述待降噪图像对应的压缩参数,并基于所述压缩参数对所述第三噪声图像进行压缩,得到第四噪声图像;Acquiring compression parameters corresponding to the image to be denoised, and compressing the third noise image based on the compression parameters to obtain a fourth noise image;
    获取所述第四噪声图像的的图像矩阵,并基于所述第四噪声图像的的图像矩阵得到所述待降噪图像的噪声分布信息。An image matrix of the fourth noise image is acquired, and noise distribution information of the image to be denoised is obtained based on the image matrix of the fourth noise image.
  15. 一种计算机可读存储介质,所述计算机可读存储介质上存储有计算机可读指令,所述计算机可读指令被处理器执行时实现如下所述的基于噪声场的图像降噪方法:A computer-readable storage medium, storing computer-readable instructions on the computer-readable storage medium, and when the computer-readable instructions are executed by a processor, implements the following noise field-based image noise reduction method:
    获取待降噪图像以及与所述待降噪图像相对应的无噪声图像;acquiring an image to be denoised and a noise-free image corresponding to the image to be denoised;
    将所述无噪声图像输入预设的噪声分布模型,并在所述噪声分布模型中构建所述无噪声图像的高斯-泊松联合噪声分布函数;Inputting the noise-free image into a preset noise distribution model, and constructing a Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model;
    根据所述高斯-泊松联合噪声分布函数获取所述待降噪图像的噪声分布信息;Acquiring noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function;
    将所述待降噪图像和所述噪声分布信息导入到预先训练好的降噪模型,根据所述噪声分布信息在以所述降噪模型中对所述待降噪图像进行降噪,得到去噪图像。Import the image to be denoised and the noise distribution information into a pre-trained denoising model, and denoise the image to be denoised in the denoising model according to the noise distribution information to obtain a denoised image. noisy image.
  16. 如权利要求15所述的计算机可读存储介质,其中,所述噪声分布模型中预设有一个高斯核和一个泊松核,所述将所述无噪声图像输入预设的噪声分布模型,并在所述噪声分布模型中构建所述无噪声图像的高斯-泊松联合噪声分布函数的步骤,具体包括:The computer-readable storage medium of claim 15, wherein a Gaussian kernel and a Poisson kernel are preset in the noise distribution model, the noise-free image is input into the preset noise distribution model, and The step of constructing the Gauss-Poisson joint noise distribution function of the noise-free image in the noise distribution model specifically includes:
    将所述无噪声图像输入到所述噪声分布模型中,通过所述噪声分布模型获取所述无噪声图像的像素信息;inputting the noise-free image into the noise distribution model, and obtaining pixel information of the noise-free image through the noise distribution model;
    基于所述无噪声图像的像素信息、所述泊松核以及所述高斯核构建所述无噪声图像的高斯-泊松联合噪声分布函数。A Gauss-Poisson joint noise distribution function of the noise-free image is constructed based on the pixel information of the noise-free image, the Poisson kernel, and the Gaussian kernel.
  17. 如权利要求16所述的计算机可读存储介质,其中,在所述将所述无噪声图像输入到所述噪声分布模型中,通过所述噪声分布模型获取所述无噪声图像的像素信息的步骤之前,还包括:The computer-readable storage medium of claim 16, wherein, in the step of inputting the noise-free image into the noise distribution model, the step of obtaining pixel information of the noise-free image through the noise distribution model Before, also included:
    从预设的图像数据库中获取样本图像,将所述样本图像输入到预设的初始噪声分布模型,获取所述初始噪声分布模型的输出结果;Obtain a sample image from a preset image database, input the sample image into a preset initial noise distribution model, and obtain an output result of the initial noise distribution model;
    构建所述初始噪声分布模型的损失函数,基于所述输出结果和预设标准结果,使用所述噪声分布模型的损失函数进行误差计算,获取识别误差;constructing a loss function of the initial noise distribution model, and using the loss function of the noise distribution model to perform error calculation based on the output result and the preset standard result to obtain an identification error;
    基于所述识别误差对所述噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型。The noise distribution model is iterated based on the identification error until the model is fitted, and the fitted noise distribution model is output.
  18. 如权利要求15所述的计算机可读存储介质,其中,所述基于所述识别误差对所述噪声分布模型进行迭代,直至模型拟合,得到输出拟合的噪声分布模型的步骤,具体包括:The computer-readable storage medium according to claim 15, wherein the step of iterating the noise distribution model based on the identification error until the model is fitted to obtain a fitted noise distribution model specifically comprises:
    将所述识别误差与预设误差阈值进行比对;comparing the identification error with a preset error threshold;
    若所述识别误差大于预设误差阈值,则基于反向传播算法对所述初始噪声分布模型进行迭代更新,直至所述识别误差小于或等于预设误差阈值为止,得到输出拟合的噪声分布模型;If the identification error is greater than a preset error threshold, the initial noise distribution model is iteratively updated based on the back-propagation algorithm until the identification error is less than or equal to the preset error threshold, and an output fitted noise distribution model is obtained ;
    输出所述噪声分布模型。The noise distribution model is output.
  19. 如权利要求9至18任意一项所述的计算机可读存储介质,其中,所述根据所述高斯-泊松联合噪声分布函数获取所述待降噪图像的噪声分布信息的步骤,具体包括:The computer-readable storage medium according to any one of claims 9 to 18, wherein the step of acquiring the noise distribution information of the image to be denoised according to the Gauss-Poisson joint noise distribution function specifically includes:
    基于所述无噪声图像的像素信息对所述高斯-泊松联合噪声分布函数进行随机抽样,得到抽样矩阵;Randomly sample the Gauss-Poisson joint noise distribution function based on the pixel information of the noise-free image to obtain a sampling matrix;
    根据所述抽样矩阵得到所述待降噪图像的噪声分布信息。The noise distribution information of the image to be denoised is obtained according to the sampling matrix.
  20. 如权利要求19所述的计算机可读存储介质,其中,所述根据所述抽样矩阵得到所述待降噪图像的噪声分布信息的步骤,具体包括:The computer-readable storage medium according to claim 19, wherein the step of obtaining the noise distribution information of the image to be denoised according to the sampling matrix specifically comprises:
    获取所述无噪声图像的图像矩阵,融合所述无噪声图像的图像矩阵和所述抽样矩阵,得到第一噪声图像;obtaining the image matrix of the noise-free image, and fusing the image matrix of the noise-free image and the sampling matrix to obtain a first noise image;
    获取所述待降噪图像对应的相机响应函数,并基于所述相机响应函数对所述第一噪声图像处理,得到第二噪声图像;acquiring a camera response function corresponding to the image to be denoised, and processing the first noise image based on the camera response function to obtain a second noise image;
    获取所述待降噪图像对应的格式转化信息,并基于所述格式转化信息对所述第二噪声图像进行彩色插值,得到第三噪声图像;acquiring format conversion information corresponding to the image to be denoised, and performing color interpolation on the second noise image based on the format conversion information to obtain a third noise image;
    获取所述待降噪图像对应的压缩参数,并基于所述压缩参数对所述第三噪声图像进行压缩,得到第四噪声图像;Acquiring compression parameters corresponding to the image to be denoised, and compressing the third noise image based on the compression parameters to obtain a fourth noise image;
    获取所述第四噪声图像的的图像矩阵,并基于所述第四噪声图像的的图像矩阵得到所述待降噪图像的噪声分布信息。An image matrix of the fourth noise image is acquired, and noise distribution information of the image to be denoised is obtained based on the image matrix of the fourth noise image.
PCT/CN2021/090430 2021-02-26 2021-04-28 Noise field-based image noise reduction method and apparatus, device, and storage medium WO2022178975A1 (en)

Applications Claiming Priority (2)

Application Number Priority Date Filing Date Title
CN202110219477.9 2021-02-26
CN202110219477.9A CN112950501B (en) 2021-02-26 2021-02-26 Noise field-based image noise reduction method, device, equipment and storage medium

Publications (1)

Publication Number Publication Date
WO2022178975A1 true WO2022178975A1 (en) 2022-09-01

Family

ID=76246622

Family Applications (1)

Application Number Title Priority Date Filing Date
PCT/CN2021/090430 WO2022178975A1 (en) 2021-02-26 2021-04-28 Noise field-based image noise reduction method and apparatus, device, and storage medium

Country Status (2)

Country Link
CN (1) CN112950501B (en)
WO (1) WO2022178975A1 (en)

Families Citing this family (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN114764803B (en) * 2022-06-16 2022-09-20 深圳深知未来智能有限公司 Noise evaluation method and device based on real noise scene and storage medium

Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN108846842A (en) * 2018-07-04 2018-11-20 武汉斗鱼网络科技有限公司 A kind of picture noise detection method, device and electronic equipment
CN109064428A (en) * 2018-08-01 2018-12-21 Oppo广东移动通信有限公司 A kind of image denoising processing method, terminal device and computer readable storage medium
CN110163827A (en) * 2019-05-28 2019-08-23 腾讯科技(深圳)有限公司 Training method, image de-noising method, device and the medium of image denoising model
CN112001856A (en) * 2020-07-29 2020-11-27 东软医疗系统股份有限公司 Training method of denoising model, image noise removing method and related device

Family Cites Families (6)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
JP2006310999A (en) * 2005-04-27 2006-11-09 Sony Corp Image processing apparatus and method, and program
CN107169932A (en) * 2017-03-21 2017-09-15 南昌大学 A kind of image recovery method based on Gauss Poisson mixed noise model suitable for neutron imaging system diagram picture
CN108564555B (en) * 2018-05-11 2021-09-21 中北大学 NSST and CNN-based digital image noise reduction method
CN109754376B (en) * 2018-12-28 2020-11-10 深圳美图创新科技有限公司 Image denoising method and device
CN110072052B (en) * 2019-04-09 2021-08-27 Oppo广东移动通信有限公司 Image processing method and device based on multi-frame image and electronic equipment
CN111405206B (en) * 2020-02-25 2022-07-26 南京邮电大学 Poisson-Gaussian combined noise image sequence separation noise reduction method

Patent Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN108846842A (en) * 2018-07-04 2018-11-20 武汉斗鱼网络科技有限公司 A kind of picture noise detection method, device and electronic equipment
CN109064428A (en) * 2018-08-01 2018-12-21 Oppo广东移动通信有限公司 A kind of image denoising processing method, terminal device and computer readable storage medium
CN110163827A (en) * 2019-05-28 2019-08-23 腾讯科技(深圳)有限公司 Training method, image de-noising method, device and the medium of image denoising model
CN112001856A (en) * 2020-07-29 2020-11-27 东软医疗系统股份有限公司 Training method of denoising model, image noise removing method and related device

Also Published As

Publication number Publication date
CN112950501B (en) 2023-10-13
CN112950501A (en) 2021-06-11

Similar Documents

Publication Publication Date Title
US11978245B2 (en) Method and apparatus for generating image
WO2022142009A1 (en) Blurred image correction method and apparatus, computer device, and storage medium
TW202213265A (en) Image processing method and device, processor, electronic equipment and storage medium
CN112418292B (en) Image quality evaluation method, device, computer equipment and storage medium
CN110189246B (en) Image stylization generation method and device and electronic equipment
WO2023035531A1 (en) Super-resolution reconstruction method for text image and related device thereof
CN112906721B (en) Image processing method, device, equipment and computer readable storage medium
WO2023159746A1 (en) Image matting method and apparatus based on image segmentation, computer device, and medium
CN114792355B (en) Virtual image generation method and device, electronic equipment and storage medium
WO2022252372A1 (en) Image processing method, apparatus and device, and computer-readable storage medium
CN117173609A (en) Multi-scale feature and channel attention-based reference-free screen video quality evaluation method and device
CN110211017A (en) Image processing method, device and electronic equipment
CN113724136A (en) Video restoration method, device and medium
CN118283297A (en) Video data processing method and device, electronic equipment and readable storage medium
CN117336527A (en) Video editing method and device
CN115358917B (en) Method, equipment, medium and system for migrating non-aligned faces of hand-painted styles
WO2022178975A1 (en) Noise field-based image noise reduction method and apparatus, device, and storage medium
CN113177483A (en) Video object segmentation method, device, equipment and storage medium
CN118042246A (en) Video generation method, device, electronic equipment and readable storage medium
CN116977195A (en) Method, device, equipment and storage medium for adjusting restoration model
CN114418835B (en) Image processing method, device, equipment and medium
CN112966150A (en) Video content extraction method and device, computer equipment and storage medium
CN115690238A (en) Image generation and model training method, device, equipment and storage medium
CN116129534A (en) Image living body detection method and device, storage medium and electronic equipment
CN113362249A (en) Text image synthesis method and device, computer equipment and storage medium

Legal Events

Date Code Title Description
121 Ep: the epo has been informed by wipo that ep was designated in this application

Ref document number: 21927408

Country of ref document: EP

Kind code of ref document: A1

NENP Non-entry into the national phase

Ref country code: DE

122 Ep: pct application non-entry in european phase

Ref document number: 21927408

Country of ref document: EP

Kind code of ref document: A1