Deep learning infrared image denoising method and system based on multi-head self-attention mechanism
An infrared image and deep learning technology, applied in the field of deep learning infrared image denoising, can solve the problems of high noise and poor imaging quality, and achieve the effect of overcoming high noise
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0070] This embodiment provides a deep learning infrared image denoising method based on a multi-head self-attention mechanism, such as figure 1 shown, including the following steps:
[0071] S1: Collect high-definition infrared images and perform preprocessing;
[0072] S2: Obtain a data set according to the high-definition infrared image and the preprocessed infrared image;
[0073] S3: Establish the infrared denoising neural network based on the multi-head self-attention mechanism, use the data set to train the infrared denoising neural network, and obtain the trained infrared denoising neural network;
[0074] S4: Use the trained infrared denoising neural network to denoise the noisy infrared image.
[0075] The multi-head self-attention mechanism was originally used in natural language processing tasks. Because of its powerful global feature extraction and local feature extraction capabilities, it has achieved good results in vision tasks instead of convolutional neural...
Embodiment 2
[0083] On the basis of embodiment 1, this embodiment specifically discloses:
[0084] The infrared denoising neural network based on the multi-head self-attention mechanism in the step S3, such as figure 2 As shown, specifically:
[0085] The infrared denoising neural network based on the multi-head self-attention mechanism includes a local feature extraction module 101, a global feature extraction module 102 and an image recovery module 103, wherein the local feature extraction module 101 extracts the local feature information I in the infrared image local , the global feature extraction module 102 extracts the global feature information I in the infrared image global , the image restoration module 103 takes the local feature information I local and global feature information I global After fusion, it is restored to a noise-free infrared image I rec .
[0086] The local feature extraction module 101, such as image 3 As shown, specifically:
[0087] The local feature ...
Embodiment 3
[0116] A deep learning infrared image denoising system based on multi-head self-attention mechanism, such as Figure 8 shown, including:
[0117] Image collection and processing module, described image collection and processing module is used for collecting high-definition infrared image and carries out preprocessing;
[0118] A data set module, the data set module is used to obtain a data set according to the high-definition infrared image and the preprocessed infrared image;
[0119] A multi-head self-attention denoising neural network module, the multi-head self-attention denoising neural network module is used to establish an infrared denoising neural network based on a multi-head self-attention mechanism, and the infrared denoising neural network is processed by using the data set Carry out training to obtain the trained infrared denoising neural network;
[0120] Denoising module, described denoising module utilizes the infrared denoising neural network trained to deno...
PUM
Abstract
Description
Claims
Application Information
- R&D Engineer
- R&D Manager
- IP Professional
- Industry Leading Data Capabilities
- Powerful AI technology
- Patent DNA Extraction
Browse by: Latest US Patents, China's latest patents, Technical Efficacy Thesaurus, Application Domain, Technology Topic, Popular Technical Reports.
© 2024 PatSnap. All rights reserved.Legal|Privacy policy|Modern Slavery Act Transparency Statement|Sitemap|About US| Contact US: help@patsnap.com