CN114723608B - Image super-resolution reconstruction method based on fluid particle network - Google Patents
Image super-resolution reconstruction method based on fluid particle network Download PDFInfo
- Publication number
- CN114723608B CN114723608B CN202210393353.7A CN202210393353A CN114723608B CN 114723608 B CN114723608 B CN 114723608B CN 202210393353 A CN202210393353 A CN 202210393353A CN 114723608 B CN114723608 B CN 114723608B
- Authority
- CN
- China
- Prior art keywords
- image
- layer
- network
- convolution
- convolution layer
- Prior art date
- Legal status (The legal status 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 status listed.)
- Active
Links
- 239000012530 fluid Substances 0.000 title claims abstract description 64
- 239000002245 particle Substances 0.000 title claims abstract description 59
- 238000000034 method Methods 0.000 title claims abstract description 39
- 238000000605 extraction Methods 0.000 claims abstract description 35
- 238000012549 training Methods 0.000 claims abstract description 30
- 238000012360 testing method Methods 0.000 claims abstract description 14
- 238000005070 sampling Methods 0.000 claims description 20
- 230000006870 function Effects 0.000 claims description 16
- 230000004913 activation Effects 0.000 claims description 11
- 238000011176 pooling Methods 0.000 claims description 9
- 238000007781 pre-processing Methods 0.000 claims description 3
- 230000003321 amplification Effects 0.000 claims description 2
- 238000009795 derivation Methods 0.000 claims description 2
- 238000003199 nucleic acid amplification method Methods 0.000 claims description 2
- 230000008569 process Effects 0.000 abstract description 18
- 230000008859 change Effects 0.000 description 7
- 230000033001 locomotion Effects 0.000 description 7
- 238000004088 simulation Methods 0.000 description 7
- 238000010586 diagram Methods 0.000 description 5
- 239000000284 extract Substances 0.000 description 4
- 230000004927 fusion Effects 0.000 description 4
- 238000003384 imaging method Methods 0.000 description 4
- 238000001514 detection method Methods 0.000 description 3
- 230000000694 effects Effects 0.000 description 3
- 238000005516 engineering process Methods 0.000 description 3
- ZXQYGBMAQZUVMI-GCMPRSNUSA-N gamma-cyhalothrin Chemical compound CC1(C)[C@@H](\C=C(/Cl)C(F)(F)F)[C@H]1C(=O)O[C@H](C#N)C1=CC=CC(OC=2C=CC=CC=2)=C1 ZXQYGBMAQZUVMI-GCMPRSNUSA-N 0.000 description 3
- 238000013507 mapping Methods 0.000 description 3
- 230000002123 temporal effect Effects 0.000 description 3
- 230000005540 biological transmission Effects 0.000 description 2
- 238000012545 processing Methods 0.000 description 2
- 238000012935 Averaging Methods 0.000 description 1
- 238000002679 ablation Methods 0.000 description 1
- 230000009471 action Effects 0.000 description 1
- 238000004458 analytical method Methods 0.000 description 1
- 238000013459 approach Methods 0.000 description 1
- 238000013528 artificial neural network Methods 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 230000000295 complement effect Effects 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 238000013461 design Methods 0.000 description 1
- 230000002708 enhancing effect Effects 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 239000011859 microparticle Substances 0.000 description 1
- 230000001537 neural effect Effects 0.000 description 1
- 238000003672 processing method Methods 0.000 description 1
- 230000001902 propagating effect Effects 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 238000012216 screening Methods 0.000 description 1
- 230000000007 visual effect Effects 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T3/00—Geometric image transformations in the plane of the image
- G06T3/40—Scaling of whole images or parts thereof, e.g. expanding or contracting
- G06T3/4053—Scaling of whole images or parts thereof, e.g. expanding or contracting based on super-resolution, i.e. the output image resolution being higher than the sensor resolution
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F17/00—Digital computing or data processing equipment or methods, specially adapted for specific functions
- G06F17/10—Complex mathematical operations
- G06F17/11—Complex mathematical operations for solving equations, e.g. nonlinear equations, general mathematical optimization problems
- G06F17/13—Differential equations
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/048—Activation functions
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
- G06T5/50—Image enhancement or restoration using two or more images, e.g. averaging or subtraction
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/10—Segmentation; Edge detection
- G06T7/11—Region-based segmentation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20112—Image segmentation details
- G06T2207/20132—Image cropping
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Mathematical Physics (AREA)
- Data Mining & Analysis (AREA)
- Software Systems (AREA)
- General Engineering & Computer Science (AREA)
- Computing Systems (AREA)
- Health & Medical Sciences (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Computational Linguistics (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- Artificial Intelligence (AREA)
- Life Sciences & Earth Sciences (AREA)
- Evolutionary Computation (AREA)
- Pure & Applied Mathematics (AREA)
- Computational Mathematics (AREA)
- Mathematical Analysis (AREA)
- Mathematical Optimization (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Operations Research (AREA)
- Algebra (AREA)
- Databases & Information Systems (AREA)
- Image Analysis (AREA)
Abstract
The invention provides an image super-resolution reconstruction method based on a fluid particle network, which comprises the following steps: acquiring a training sample set and a testing sample set; (2) Constructing an image reconstruction model O based on the fluid particle network; (3) Performing iterative training on the fluid particle network reconstruction network model O; and (4) acquiring an image reconstruction result. In the process of training the fluid particle network model and acquiring the image reconstruction result, the mixed second-order residual extraction network can effectively learn the representative features, and the branch network is guided to realize the adjustment of the image super-resolution by guiding the pixel moving direction, so that clearer detail and parameter performance can be obtained, and the resolution of the reconstructed image is effectively improved.
Description
Technical Field
The invention belongs to the technical field of image processing, relates to an image super-resolution reconstruction method, and particularly relates to an image super-resolution reconstruction method based on a fluid particle network, which can be used in the technical fields of hardware Trojan horse detection technology and the like.
Background
During the imaging process, due to the influence of various factors of the imaging system, the obtained image may not be a perfect image of the real scene. The image reconstruction is to reconstruct a degraded image to restore the original appearance of a scene to the maximum extent, and the image reconstruction can only make the image approach to the original image as much as possible, but is difficult to accurately restore due to factors such as noise interference. Due to the limitation of an imaging system or transmission bandwidth, the resolution of the obtained image is often low, and the image super-resolution reconstruction (SR) is to reconstruct a higher-resolution image by using the existing image. In the field of strict imaging quality requirements of technologies such as hardware Trojan horse detection and the like, the image is required to have higher resolution, and the image structure should not have structural distortion and edge texture errors, so that the target is clear and is easy to identify. And measuring the quality index of the reconstructed image by adopting the peak signal-to-noise ratio and the structural similarity. In practical situations, most image super-resolution reconstruction networks learn mapping from an LR image to an HR image through an improved residual learning idea, and increase the number of layers to improve the reconstruction capability of the network, the image super-resolution reconstruction networks only distribute weights for different mappings through a trapezoidal formula, and the SR task is researched from the aspect of image evolution in the process of a fluid particle network SR, which is very important for establishing an interpretable and effective model, and meanwhile, a potential research direction is provided.
Shenzhen saolu medical science and technology Limited in its patent document, "super-resolution image reconstruction method and electronic device with stepwise fusion features" (patent application No. 202110842925.0, application publication No. CN 111461983A), proposes a super-resolution image reconstruction method with stepwise fusion features, which includes: the method comprises the steps of obtaining an original image and a reconstruction network, utilizing a first feature extraction module to extract features of the original image, enabling the original feature map to sequentially pass through a plurality of second feature extraction modules and a plurality of third feature extraction modules, utilizing an image reconstruction module to carry out super-resolution reconstruction on the intermediate feature map to obtain a target image with higher resolution, and through multiple feature fusion, gradually screening out features useful for super-resolution reconstruction of the image for multiple times and finally inputting the features into the image reconstruction module. Although the resolution of the reconstructed image is improved by the method, the driving force of the reconstruction network is insufficient along with the increase of the number of the network layers, and the further improvement of the resolution of the reconstructed image of the neural network is limited.
Disclosure of Invention
The invention aims to overcome the defects of the prior art and provides an image super-resolution reconstruction method based on a fluid particle network, aiming at enabling a reconstructed image to have better edge detail reconstruction effect and parameter performance by improving the driving force of the fluid particle network.
In order to achieve the purpose, the technical scheme adopted by the invention comprises the following steps:
1. an image super-resolution reconstruction method based on a fluid particle network is characterized by comprising the following steps:
(1) Acquiring a training sample set and a testing sample set:
(1a) Acquiring K RGB images, preprocessing each RGB image, and performing 1/4 down-sampling on each processed image to obtain K RGB images after down-sampling;
(1b) Cutting each RGB image into image blocks with the size of H multiplied by H, and simultaneously cutting the down-sampled RGB image corresponding to the RGB image into image blocks with the size of H multiplied by HThe image blocks cut from each RGB image are used as labels of the corresponding cut image blocks after down-sampling, and then more than half of the cut image blocks after down-sampling and the labels thereof are randomly selected to form a training sample set R 1 Combining the residual down-sampled image blocks and labels thereof into a test sample set E 1 Wherein K is more than or equal to 600, H is more than or equal to 256;
(2) Constructing a fluid particle network model O:
constructing a fluid particle network model O comprising a first convolution layer, D fluid particle sub-networks and an up-sampling module which are connected in sequence; each fluid particle sub-network comprises a guide branch network and a mixed second-order residual error network which are arranged in parallel, and D is more than or equal to 10; the guiding branch network comprises a feature compensation layer, a feature expansion block and a plurality of convolution layers, wherein the feature expansion block comprises the plurality of convolution layers which are sequentially cascaded; the mixed second-order residual error network comprises a characteristic compensation layer, a mixed second-order residual error extraction layer, a channel attention module and a plurality of convolution layers, wherein the channel attention module comprises a pooling layer, a plurality of convolution layers and a plurality of nonlinear activation layers;
(3) Performing iterative training on the fluid particle network model O:
(3a) The number of initial iterations is S, the maximum number of iterations is S, S is more than or equal to 10000, and the model of the fluid particle network of the S-th iteration is O s ,O s The weight and bias parameters of the middle learnable parameter are w respectively s 、b s And let s =1,O s =O;
(3b) Will be derived from the training sample set R 1 Selecting N training samples as a fluid particle network model O at random and with replacement s The method comprises the steps that a first convolution layer carries out feature extraction on each training sample to obtain N feature maps, a guide branch network in a fluid particle sub-network carries out image reconstruction on each feature map to obtain N compensation images, meanwhile, a mixed second-order residual error network carries out image reconstruction on each feature map to obtain N intermediate images, each compensation image and the intermediate image of the corresponding position of the compensation image are added element by element and repeated for D times to obtain N images containing rich details and edge information, and a module carries out up-sampling on each image containing rich details and edge information to obtain N reconstructed images, wherein N is larger than or equal to 600;
(3c) Calculating a loss function by using the L1 norm, and calculating O by using the reconstructed image and the corresponding training sample label s Loss value L of s Separately calculating L by the chain rule s For weight parameter omega in network s And bias parameter b s Partial derivatives ofAndand according to >>For omega s 、b s And (6) updating.
(3d) Judging whether S is greater than or equal to S, if so, obtaining a trained fluid particle network model O, otherwise, enabling S = S +1, and executing the step (3 b);
(4) Acquiring an image reconstruction result:
set of test samples E 1 And performing forward propagation as an input of the trained fluid particle network model O to obtain a reconstructed image corresponding to all the test samples.
Compared with the prior art, the invention has the following advantages:
the fluid particle network model constructed by the invention comprises a plurality of fluid particle sub-networks, each fluid particle sub-network comprises a guide branch network and a mixed second-order residual error network which are arranged in parallel, the mixed second-order residual error extraction network can effectively learn representative characteristics in the process of training the model and reconstructing an image, the guide branch network can realize the adjustment of image super-resolution by guiding the pixel moving direction, the clearer detail and parameter performance can be obtained, and experimental results show that the invention can effectively improve the resolution of natural image and real character microchip image reconstruction.
Drawings
FIG. 1 is a flow chart of an implementation of the present invention;
FIG. 2 is a schematic diagram of a model of a fluid particle network according to the present invention;
FIG. 3 is a schematic diagram of a network structure of a guided branch according to an embodiment of the present invention;
fig. 4 is a schematic diagram of a residual second-order extraction network structure according to an embodiment of the present invention.
Detailed Description
The invention is described in further detail below with reference to the figures and specific examples.
Referring to fig. 1, the present invention includes the steps of:
step 1) obtaining a training sample set R 1 And test sample set E 1 :
Step 1 a) acquiring K RGB images from a DIV2K data set, wherein K is more than or equal to 600. In the present embodiment, K =600;
step 1 b) subtracting the average gray value of each channel of each RGB image in the DIV2K data set from the value of each channel of each RGB image in the K RGB images to obtain each RGB image, and then randomly horizontally turning and rotating by 90 degrees to perform data enhancement on each RGB image. And performing 1/4 down-sampling on each processed image to obtain K RGB images after down-sampling.
The method comprises the following implementation steps: suppose that the image value of the RGB channel of the first image is [ R ] 1 ,G 2 ,B 3 ]Averaging all the pixels on R, G and B channels of K RGB images respectively, and the value is [ R, G, B ]]The image value on the R, G and B channels of the processed first RGB image is [ R ] 1 ,G 2 ,B 3 ]-[r,g,b]And (3) carrying out the processing on each RGB image, subtracting the average value from each RGB image, highlighting the difference of each RGB image, then carrying out data enhancement by randomly horizontally overturning and rotating by 90 degrees each RGB image, increasing the data amount of training, and carrying out 1/4 down-sampling on each processed image to obtain K down-sampled RGB images.
Step 1 b) cutting each RGB image into image blocks with the size of H multiplied by H, and simultaneously cutting the corresponding down-sampled RGB image of the RGB image into image blocks with the size of H multiplied by HThe image blocks of each RGB image are used as labels of the corresponding cut image blocks after downsampling, and then more than half of the cut image blocks after downsampling and the labels are randomly selected to form a training sample set R 1 Combining the image blocks and labels thereof after residual down-sampling into a test sample set E 1 。
Step 2) constructing an image reconstruction model O based on the fluid particle network, wherein the structure of the image reconstruction model O is shown in FIG. 2;
constructing a fluid particle network model O comprising a first convolution layer, D fluid particle sub-networks and an up-sampling module which are connected in sequence; each fluid particle sub-network comprises a guide branch network and a mixed second-order residual network which are arranged in parallel, D is more than or equal to 10, and in the embodiment, D =10; the guiding branch network comprises a feature compensation layer, a feature expansion block and a plurality of convolution layers, wherein the feature expansion block comprises the plurality of convolution layers which are sequentially cascaded; the mixed second-order residual error main feature extraction network comprises a feature compensation layer, a mixed second-order residual error extraction layer, a channel attention module and a plurality of convolution layers, wherein the channel attention module comprises a pooling layer, a plurality of convolution layers and a plurality of nonlinear activation layers;
the structure of the guided branch network in this embodiment is shown in fig. 3;
the guiding branch network comprises a feature compensation layer, a second convolution layer, a feature expansion block and a third convolution layer which are sequentially cascaded, wherein the expansion block comprises a fourth convolution layer, a fifth convolution layer and a sixth convolution layer which are sequentially cascaded; the network parameters are set as: the sizes of the second convolution layer and the fourth convolution layer convolution kernel are 3 × 3, the number of channels is set to 64, and the size of the third convolution layer convolution kernel is 1 × 1, the number of channels is set to 64; the fifth convolution layer and the sixth convolution layer are convolution layers with two step lengths of 1 and 2 respectively, and the number of channels is set to be 64;
characteristic compensation layer: inspired by the theory related to fluid mechanics. By studying the similarity between the motion of the fluid Field (FD) microparticles and the image pixels in the image field (SR). Specifically, the movement of the pixel during SR is similar to the fluid flow in FD. This similarity is in two respects. From a microscopic perspective, the elementary cell particles studied by FD move under the combined action of temperature, pressure and surrounding particles. And the pixels in the image change in a targeted manner under the influence of operations such as convolution and pooling, and surrounding pixels. From a macroscopic point of view, the distribution of the same cross section in the fluid varies regularly over time. In the SR process, after multiple times of feature extraction and reconstruction, the two-dimensional feature mapping changes towards the target HR image. The image SR reconstruction process can be regarded as a fluid particle motion process. The N-S equation in FD establishes the relationship between various properties of the particle in time and space, which can transform different information into each other. The N-S equation may be applied to the image SR process to convert the image.
The structural boundaries of HR images obtained due to the normal cnn network are not clear. To address this problem, additional constraints are used to constrain the SR process. The residual extraction process is mainly designed from the perspective of structural features, and besides the structural features, the change rate of pixel values with time is another important information. The lead branch is designed with consideration for guiding the residue extraction process from the perspective of dynamic pixel value distribution.
The guide branch inspired by the N-S equation: the continuity equation in FD can be divided into four forms, i.e., integral equation, partial differential equation, conservative type, and non-conservative type. Since they can be transformed into each other, a partial differential equation in a conservative form is taken as an example:
where p is the density of the fluid,is a velocity vector, is greater than or equal to>Using the multiplication law of hamiltonian and vector functions, equation (8) can be rewritten as:
in the SR process, the extracted feature map may change as the depth of the network changes. This process can be viewed as a change in cross-section in a three-dimensional fluid, and thus the equation of motion of the three-dimensional flow field is derived as an equation of motion in a two-dimensional image domain, where the law of motion of the particles can be expressed as:
it can be rewritten as:
the variation in divergence of (d) is a variation in the number of pixels per unit area. In the process of image reconstruction, pixels of different characteristic regions cannot flow in or out without fusion of a plurality of images. Thus it is assumed that>I.e. the partial derivatives of the velocity components u and w in the respective directions are 0. Equation (11) can then be rewritten as:
in the image domain, ρ represents a pixel value. Partial derivative ofAnd &>Representing the gradient of the pixel value in different directions. For example, the edge area has a larger gradation, while the smooth area has a smaller gradation. The target variable in equation (12) is a partial derivative with time, which can be expressed in differential form:
the relationship between the temporal and spatial dimensions is established in the feature extraction process. Since the two-dimensional partial derivatives are calculated using the neural layer, equation (13) can be rewritten as:
and the branches are guided to apply the time and space relation in the N-S control equation to feature extraction and reconstruction under the inspiration of the N-S control equation. The N-S equation can be converted to the form of equation (12) where the first term is the rate of change of the density function in the time dimension and the latter two terms are the rate of change of the density function in the space dimension. The spatial information (in x-direction and y-direction) can be converted into temporal information (in i-direction) using equation (14), thereby enabling the guidance of the skeleton extraction result and obtaining more details. According to the density information and the two-dimensional partial derivative at a certain moment, approximate density distribution information at the next moment can be obtained. The change rate of the density function in the spatial dimension can be used as supplementary characteristic information for predicting the pixel point distribution at the next moment so as to guide the SR process.
The structure of the hybrid second-order residual error network of the embodiment is shown in fig. 4;
the specific structure of the hybrid second-order residual error network comprises two branches which are arranged in parallel, the specific structure of the first branch is a characteristic compensation layer, the specific structure of the second branch is a seventh convolution layer, an eighth convolution layer, a ninth convolution layer, a tenth convolution layer, an eleventh convolution layer, a hybrid second-order residual error extraction layer and a channel attention module which are sequentially cascaded, and the output end of the characteristic compensation layer is connected with the input end of the hybrid second-order residual error extraction layer; the channel attention module comprises a pooling layer, a twelfth convolution layer, a first nonlinear activation layer, a thirteenth convolution layer and a second nonlinear activation layer which are sequentially cascaded; the network parameters are set as: the eighth convolution layer and the tenth convolution layer have a step size of 2, the seventh convolution layer, the ninth convolution layer and the eleventh convolution layer have a step size of 1/2, the number of channels is set to 64, the convolution kernel size of the twelfth convolution layer and the thirteenth convolution kernel size is 1 × 1, the number of channels is set to 32, the pooling layer is set to be maximally pooled, the first nonlinear active layer is realized by a LeakyReLU function, and the second nonlinear active layer is realized by a Sigmoid function.
Mixing a second-order residual extraction layer: in the fluid Field (FD), the position of the particles changes with time, and the cut surface of the fluid exhibits different distribution images at different times; similarly, the SR process can be thought of as a series of temporally evolving intermediate pictures; because of the similarity of element motion in the fluid field and the image field, a difference equation in the fluid field is used for describing the process of the image SR, a first-order expression in a finite difference equation is the same as that of a classical residual error network, a finite difference equation system is introduced into the design of the network, more relationships are established among characteristic items by a high-order difference equation, second-order mixed derivatives in the finite difference equation can interact with each other and extract information of different dimensions, and a second-order mixed structure is used for designing a network structure; a new residual structure is proposed according to a second-order hybrid differential equation (an extended two-dimensional differential operation); setting the dimension of j +1= m as the time direction, namely a multi-feature extraction process; j-1= n dimension is set for guiding the branch network, and the second-order mixed differential equation is deformed; and obtaining a residual subtraction form, wherein the residual of the residual is the information needing to be learned in the network. In this configuration, the nth residual is the difference between the characteristics of the n-1 and n-2 layers.
The second order mixing difference is defined as:
where Δ x and Δ y are constants representing the step size in their respective directions.
The x-direction is defined as the temporal direction, which describes the process of feature extraction. Meanwhile, the y direction is defined as a plurality of image data processing methods. One dimension (j +1= m) extracts the global features using the convolutional layer, and the other dimension (j-1 = n) extracts the global features using the convolutional layer. And extracting detailed compensation information by using the guide branch network. These two-dimensional information may complement each other in terms of feature extraction and transmission. Based on this setup, the equation is converted to the following form:
wherein m represents the extraction of the main features of the mixed second-order residual error network, and n represents the extraction of the supplementary features by using the guide branch network.
To simplify equation (16), the step term is replaced with h and the equation is further derived as follows:
wherein u is i+1,m And u i-1,m And (3) representing the extraction result of the mixed second-order residual error network backbone structure at i +1 and i-1. u. of i+1,n -u i-1,n Is the feature information extracted by the guide branch network, and can be expressed by delta u i+1 And (4) showing. Thus:
Δu i+1 is the feature information that guides the extraction of the branch network,is a hybrid second order residual network to learn and simplify to a two-dimensional function f (x, y) i,j The residual error of (2) is obtained by fusing the results of the global and compensation features of the convolutional layer. The residual structure based on the second order hybrid difference is finally expressed as:
u i+1,m =u i-1,m +Δu i+1 +hf(x,y) i,j (19)
hybrid second order residual error network extraction result u i-1,m +hf(x,y) i,j And guiding the extraction result delta u of the branch network i+1 Fuse in the y-direction and then follow the u-direction along the x-direction i-1,m To u i+1,m Propagating forward.
A network is constructed using correlations between multiple terms of a higher order difference equation such that each layer in the network receives information from more layers. The loss of image information in the process of feature extraction is effectively reduced, and a network structure can obtain stronger feature extraction capability.
And sending the final reconstruction result to an up-sampling module, wherein the up-sampling module is realized by PixelShuffle, and the amplification parameter is set to be 4.
Step 3) performing iterative training on the image super-resolution reconstruction network based on the fluid particle network;
step 3 a) the number of initialization iterations is S, the maximum number of iterations is S, S is more than or equal to 10000, and the model of the fluid particle network of the S-th iteration is O s ,O s The weight and bias of the middle learnable parameter are w s b s And let s =1,O s =O;
Step 3 b) will be from the training sample set R 1 Selecting N training samples as a fluid particle network model O at random and with replacement s The input of (1);
step 3 c) the first convolution layer performs convolution on the input image to obtain a feature map Y of the input image 1 Wherein f is m Representing a first feature map after convolution of the mth training sample; guiding the branch network to map each feature f m Sequentially feeding the feature compensation layer to obtain a compensation feature map Y 2 Wherein f is m ' representing a compensation feature map after convolution of the mth training sample, wherein the compensation feature map optimizes and enhances features; compensated feature map Y 2 Sending into the second convolution layer to obtain a depth feature map Y 3 A depth feature map Y 3 Sending into a feature extension block, recovering more detail information of the image through a fourth convolution layer and two convolution layers with step length of 1 and 2 respectively to obtain a depth feature map Y 4 The third convolution layer converts the depth feature map Y into 4 Is projected to 64 to obtain a depth feature map Y of the compensated image 5 。
Characteristic diagram Y 1 Obtaining m depth feature maps Y containing rich context interdependent information through a seventh convolution layer with a step length of 1/2, an eighth convolution layer with a step length of 2, a ninth convolution layer with a step length of 1/2, a tenth convolution layer with a step length of 2 and an eleventh convolution layer with a step length of 1/2 in a mixed second-order residual error network 7 Feature map Y 1 Obtaining a depth feature map Y through a feature compensation layer 8 The depth feature map Y 7 、Y 8 Sending the depth feature map into a mixed second-order residual extraction layer to obtain a depth feature map Y 9 Depth feature map Y 9 The feature graph and the compensation feature graph of the hybrid second-order residual error network are fused to obtain more detailed information in time and space dimensions, the channel attention module of the hybrid second-order residual error network can distribute different weights for different channel feature graphs, information contained in the feature graphs can be more accurate, and the intermediate image feature graph Y is obtained 10 Compensating the image depth feature map Y 5 And intermediate image depth feature map Y 10 Adding the channels element by element to obtain a characteristic diagramThe resulting characteristic map is evaluated after 10 fluid sub-networks>And obtaining a reconstructed image through an up-sampling layer.
Step 3 d) calculating a loss function by adopting the L1 norm, and calculating O through the reconstructed image and the corresponding training sample label thereof s Loss value L of s Separately calculating the loss function L by the chain rule s And according toFor omega s 、b s The updating formulas for updating are respectively as follows:
wherein,representing the reconstructed image, I representing the labels of the samples in the training sample set, w s 、b s Represents O s Weight, bias parameter, w of all learnable parameters s '、b s ' indicates updated learnable parameters,/ r Indicates the learning rate, L s Is a loss function, <' > is>Representing the derivation operation.
Step 3 e) judging whether S is more than or equal to S, if so, obtaining a trained fluid particle network model O, otherwise, making S = S +1, and executing the step (3 c);
step 3 f), obtaining an image reconstruction result:
set of test samples E 1 And performing forward propagation as an input of the trained fluid particle network model O to obtain a reconstructed image corresponding to all the test samples.
The hybrid second-order residual error network provided by the invention can effectively learn representative characteristics, and the guide branch network adjusts the SR process by guiding the pixel moving direction, thereby being beneficial to optimizing and enhancing the characteristics and obtaining clear details. Experiments on benchmarking of both the natural image SR and the real text microchip image SR verify the validity of the proposed particle network. It is superior to representative methods in both objective index and visual quality. The fluid particle network provided by the invention provides great potential for practical applications such as hardware Trojan horse detection and the like.
The technical effect of the invention is further explained by combining simulation experiments.
1. Simulation conditions and contents:
the hardware platform of the simulation experiment is as follows: the processor is an Intel (R) Core i9-9900K CPU, the main frequency is 3.6GHz, the memory is 32GB, and the display card is NVIDIA GeForce RTX 2080Ti. The software platform of the simulation experiment is as follows: ubuntu 16.04 operating system, python version 3.7, pytrch version 1.7.1.
The RGB image dataset used in the simulation experiment was the DIV2K dataset. In the simulation experiment, 800 RGB images are selected to form a data set, and each target class is selected from the data set to form a data set R corresponding to the RGB images 0 R after pretreatment 0 Form a training sample set R 1 The RGB images of the Set5 dataset and the BSD100 dataset constitute a test sample Set E 1 。
The fluid particle network proposed by the present invention consists of two parts: a hybrid second order residual network and a guided branch network. To verify the effectiveness of these two network structures, an ablation study was conducted. The results are shown in Table 1. The peak signal-to-noise ratios of the prior art on the Set5 data Set and the BSD100 data Set were 32.33dB and 27.65dB, respectively, and the peak signal-to-noise ratios of the invention on the Set5 data Set were 32.54dB and 27.71dB, respectively, with the results shown in tables 2 and 3. Compared with the prior art, the peak signal-to-noise ratio is obviously improved.
TABLE 1
TABLE 2
Prior Art | The invention | |
PSNR | 32.33dB | 32.54dB |
SSIM | 0.8970 | 0.8979 |
TABLE 3
Prior Art | The invention | |
PSNR | 27.65dB | 27.71dB |
SSIM | 0.7388 | 0.7422 |
The invention provides a method of a hybrid second-order residual error structure improved at the system level by combining the result analysis in the simulation experiment, and the pixel value of the edge area in the result of the hybrid second-order residual error network may fluctuate. Under the guidance of the guidance branch network, the edge becomes sharper. The guide branch network only extracts the derivative of the features from each layer, so that the cost is not too high, the problem of insufficient driving force of the traditional super-resolution image reconstruction method can be effectively solved, the edge detail reconstruction effect and the parameter performance are better, and the problems of loss of edge information, fuzzy texture details and structural distortion of the reconstructed image are further solved.
Claims (4)
1. An image super-resolution reconstruction method based on a fluid particle network is characterized by comprising the following steps:
(1) Acquiring a training sample set and a testing sample set:
(1a) Acquiring K RGB images, preprocessing each RGB image, and performing 1/4 down-sampling on each processed image to obtain K RGB images after down-sampling;
(1b) Cutting each RGB image into image blocks with the size of H multiplied by H, and simultaneously cutting the down-sampled RGB image corresponding to the RGB image into image blocks with the size of H multiplied by HThe image blocks cut from each RGB image are used as labels of the corresponding cut image blocks after down-sampling, and then more than half of the cut image blocks after down-sampling and the labels thereof are randomly selected to form a training sample set R 1 Combining the residual down-sampled image blocks and labels thereof into a test sample set E 1 Wherein K is more than or equal to 600, H is more than or equal to 256;
(2) Constructing a fluid particle network model O:
constructing a fluid particle network model O comprising a first convolution layer, D fluid particle sub-networks and an up-sampling module which are connected in sequence; each fluid particle sub-network comprises a guide branch network and a mixed second-order residual error network which are arranged in parallel, and D is more than or equal to 10; the guiding branch network comprises a feature compensation layer, a feature expansion block and a plurality of convolution layers, wherein the feature expansion block comprises the plurality of convolution layers which are sequentially cascaded; the mixed second-order residual error network comprises a characteristic compensation layer, a mixed second-order residual error extraction layer, a channel attention module and a plurality of convolution layers, wherein the channel attention module comprises a pooling layer, a plurality of convolution layers and a plurality of nonlinear activation layers;
(3) Performing iterative training on the fluid particle network model O:
(3a) The number of initial iterations is S, the maximum number of iterations is S, S is more than or equal to 10000, and the model of the fluid particle network of the S-th iteration is O s ,O s The weight and bias parameters of the middle learnable parameter are w respectively s 、b s And let s =1,O s =O;
(3b) Will be derived from the training sample set R 1 Selecting N training samples as a fluid particle network model O at random and with replacement s The method comprises the steps that a first convolution layer carries out feature extraction on each training sample to obtain N feature maps, a guide branch network in a fluid particle sub-network carries out image reconstruction on each feature map to obtain N compensation images, meanwhile, a mixed second-order residual error network carries out image reconstruction on each feature map to obtain N intermediate images, each compensation image and the intermediate image at the corresponding position of the compensation image are added element by element, the adding is repeated for D times to obtain N images containing rich details and edge information, and a module carries out up-sampling on each image containing the rich details and the edge information to obtain N reconstructed images, wherein N is larger than or equal to 600;
(3c) Calculating a loss function by using the L1 norm, and calculating O by using the reconstructed image and the corresponding training sample label s Loss value L of s Separately calculating L by the chain rule s For weight parameter omega in network s And bias parameter b s Partial derivatives ofAnd &>And according to >>For omega s 、b s Updating is carried out;
(3d) Judging whether S is more than or equal to S, if so, obtaining a trained fluid particle network model O, otherwise, enabling S = S +1, and executing the step (3 b);
(4) Acquiring an image reconstruction result:
set of test samples E 1 And performing forward propagation as an input of the trained fluid particle network model O to obtain a reconstructed image corresponding to all the test samples.
2. The method for super-resolution image reconstruction based on fluid particle networks according to claim 1, wherein the preprocessing is performed on each RGB image in step (1 a), specifically: the average gray value of each channel in the DIV2K dataset is subtracted from each RGB image, followed by random horizontal flipping and 90 rotations for data enhancement.
3. The method for super-resolution image reconstruction based on fluid particle networks according to claim 1, wherein the fluid particle network model O in step (2) comprises 10 fluid particle networks, and wherein:
the number of the convolution layers is 3, the number of the convolution layers contained in the characteristic extension block is 3, and the specific structure of the guiding branch network is as follows: the device comprises a feature compensation layer, a second convolution layer, a feature expansion block and a third convolution layer which are sequentially cascaded, wherein the expansion block comprises a fourth convolution layer, a fifth convolution layer and a sixth convolution layer which are sequentially cascaded; the network parameters are set as: the sizes of the convolution kernels of the first convolution layer, the second convolution layer and the fourth convolution layer are 3 × 3, the number of channels is set to 64, and the size of the convolution kernel of the third convolution layer is 1 × 1, and the number of channels is set to 64; the fifth convolution layer and the sixth convolution layer are convolution layers with two step lengths of 1 and 2 respectively, and the number of channels is set to be 64;
mixing a second-order residual error network, wherein the number of the convolutional layers is 5, the number of the convolutional layers contained in the channel attention module is 2, and the number of the nonlinear activation layers is 2; the specific structure of the hybrid second-order residual error network comprises two branches which are arranged in parallel, the specific structure of the first branch is a characteristic compensation layer, the specific structure of the second branch is a seventh convolution layer, an eighth convolution layer, a ninth convolution layer, a tenth convolution layer, an eleventh convolution layer, a hybrid second-order residual error extraction layer and a channel attention module which are sequentially cascaded, and the output end of the characteristic compensation layer is connected with the input end of the hybrid second-order residual error extraction layer; the channel attention module comprises a pooling layer, a twelfth convolution layer, a first nonlinear activation layer, a thirteenth convolution layer and a second nonlinear activation layer which are sequentially cascaded; the network parameters are set as: the convolution kernel size of the seventh convolution layer is 3 x 3, the number of channels is set to be 64, the steps of the eighth convolution layer and the tenth convolution layer are 2, the steps of the ninth convolution layer and the eleventh convolution layer are 1/2, the number of channels of the second nonlinear activation layer is set to be 64, the convolution kernel size of the twelfth convolution layer and the thirteenth convolution layer is 1 x 1, the number of channels of the twelfth convolution layer and the thirteenth convolution layer is set to be 32, the pooling layer is set to be maximum pooling, the first nonlinear activation layer is realized by a LeakyReLU function, and the second nonlinear activation layer is realized by a Sigmoid function;
the upsampling module is implemented by PixelShuffle, and the amplification parameter is 4.
4. The method for super-resolution image reconstruction based on fluid particle network as claimed in claim 1, wherein the L1 norm computation loss function L in step (3 c) s And according to For omega s 、b s The updating formulas for updating are respectively as follows:
wherein,representing the reconstructed image, I representing the label of the sample in the training sample set, w s 、b s Represents O s Weight, bias parameter, w of all learnable parameters s '、b s ' indicates updated learnable parameters,/ r Indicates the learning rate, L s Is a loss function, <' > is>Representing the derivation operation. />
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210393353.7A CN114723608B (en) | 2022-04-14 | 2022-04-14 | Image super-resolution reconstruction method based on fluid particle network |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210393353.7A CN114723608B (en) | 2022-04-14 | 2022-04-14 | Image super-resolution reconstruction method based on fluid particle network |
Publications (2)
Publication Number | Publication Date |
---|---|
CN114723608A CN114723608A (en) | 2022-07-08 |
CN114723608B true CN114723608B (en) | 2023-04-07 |
Family
ID=82243745
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210393353.7A Active CN114723608B (en) | 2022-04-14 | 2022-04-14 | Image super-resolution reconstruction method based on fluid particle network |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114723608B (en) |
Families Citing this family (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116257852B (en) * | 2023-03-21 | 2023-11-14 | 西安电子科技大学 | Chip hardware Trojan horse detection method based on differential curvature |
CN116228546B (en) * | 2023-05-04 | 2023-07-14 | 北京蔚领时代科技有限公司 | Image superdivision method and system based on channel recombination |
CN117522687B (en) * | 2023-11-03 | 2024-05-14 | 西安电子科技大学 | Super-resolution reconstruction method of hyperspectral image based on particle dynamics |
CN117541471B (en) * | 2023-11-09 | 2024-06-07 | 西安电子科技大学 | SPH heuristic PG-SPECT image super-resolution reconstruction method |
Citations (8)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN109146792A (en) * | 2018-09-25 | 2019-01-04 | 西安电子科技大学 | Chip image super resolution ratio reconstruction method based on deep learning |
CN109669049A (en) * | 2019-02-01 | 2019-04-23 | 浙江大学 | A kind of particle image velocimetry method based on convolutional neural networks |
CN111192200A (en) * | 2020-01-02 | 2020-05-22 | 南京邮电大学 | Image super-resolution reconstruction method based on fusion attention mechanism residual error network |
EP3709255A1 (en) * | 2017-11-09 | 2020-09-16 | Boe Technology Group Co. Ltd. | Image processing method, processing apparatus and processing device |
CN112288714A (en) * | 2020-10-28 | 2021-01-29 | 西安电子科技大学 | Hardware Trojan horse detection method based on deep learning |
CN113139989A (en) * | 2021-06-22 | 2021-07-20 | 浙江大学 | Particle image velocity measurement method and device based on deep learning |
CN113496499A (en) * | 2021-04-26 | 2021-10-12 | 上海理工大学 | Low-quality particle velocity field image correction method based on convolutional neural network |
WO2021246968A1 (en) * | 2020-06-05 | 2021-12-09 | National University Of Singapore | Deep fluorescence imaging by laser-scanning excitation and artificial neural network processing |
Family Cites Families (6)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN107169927B (en) * | 2017-05-08 | 2020-03-24 | 京东方科技集团股份有限公司 | Image processing system, method and display device |
US20190277989A1 (en) * | 2018-03-08 | 2019-09-12 | Chevron U.S.A. Inc. | System and method for improving resolution of digital seismic images |
WO2020096724A1 (en) * | 2018-11-09 | 2020-05-14 | University Of Florida Research Foundation, Inc. | Histogram-based method for auto segmentation of integrated circuit structures from sem images |
US11507662B2 (en) * | 2019-02-04 | 2022-11-22 | Sateesh Kumar Addepalli | Systems and methods of security for trusted artificial intelligence hardware processing |
EP3991096A1 (en) * | 2019-06-26 | 2022-05-04 | HRL Laboratories, LLC | System and method fdr detecting backdoor attacks in convolutional neural networks |
US11394742B2 (en) * | 2020-08-17 | 2022-07-19 | International Business Machines Corporation | Detecting trojan neural networks |
-
2022
- 2022-04-14 CN CN202210393353.7A patent/CN114723608B/en active Active
Patent Citations (8)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
EP3709255A1 (en) * | 2017-11-09 | 2020-09-16 | Boe Technology Group Co. Ltd. | Image processing method, processing apparatus and processing device |
CN109146792A (en) * | 2018-09-25 | 2019-01-04 | 西安电子科技大学 | Chip image super resolution ratio reconstruction method based on deep learning |
CN109669049A (en) * | 2019-02-01 | 2019-04-23 | 浙江大学 | A kind of particle image velocimetry method based on convolutional neural networks |
CN111192200A (en) * | 2020-01-02 | 2020-05-22 | 南京邮电大学 | Image super-resolution reconstruction method based on fusion attention mechanism residual error network |
WO2021246968A1 (en) * | 2020-06-05 | 2021-12-09 | National University Of Singapore | Deep fluorescence imaging by laser-scanning excitation and artificial neural network processing |
CN112288714A (en) * | 2020-10-28 | 2021-01-29 | 西安电子科技大学 | Hardware Trojan horse detection method based on deep learning |
CN113496499A (en) * | 2021-04-26 | 2021-10-12 | 上海理工大学 | Low-quality particle velocity field image correction method based on convolutional neural network |
CN113139989A (en) * | 2021-06-22 | 2021-07-20 | 浙江大学 | Particle image velocity measurement method and device based on deep learning |
Also Published As
Publication number | Publication date |
---|---|
CN114723608A (en) | 2022-07-08 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN114723608B (en) | Image super-resolution reconstruction method based on fluid particle network | |
CN109035142B (en) | Satellite image super-resolution method combining countermeasure network with aerial image prior | |
CN108765282B (en) | Real-time super-resolution method and system based on FPGA | |
Ren et al. | PhySR: Physics-informed deep super-resolution for spatiotemporal data | |
KR102543690B1 (en) | Image Upscaling Apparatus And Method Based On Learning With Privileged Information | |
Kim et al. | Pynet-ca: enhanced pynet with channel attention for end-to-end mobile image signal processing | |
Ma et al. | A super-resolution convolutional-neural-network-based approach for subpixel mapping of hyperspectral images | |
Qin et al. | Lightweight single image super-resolution with attentive residual refinement network | |
Sharma et al. | An efficient image super resolution model with dense skip connections between complex filter structures in Generative Adversarial Networks | |
Piras et al. | Fast and realistic large-scale structure from machine-learning-augmented random field simulations | |
Benfenati et al. | Neural blind deconvolution with Poisson data | |
Liu et al. | Gradient prior dilated convolution network for remote sensing image super-resolution | |
Tang et al. | Single-frame super-resolution for remote sensing images based on improved deep recursive residual network | |
Gao et al. | Lightweight image super-resolution via multi-branch aware CNN and efficient transformer | |
Ghasemi-Falavarjani et al. | Particle filter based multi-frame image super resolution | |
Wang et al. | A CBAM‐GAN‐based method for super‐resolution reconstruction of remote sensing image | |
Liu et al. | Fine-grained scale space learning for single image super-resolution | |
Franchi et al. | Supervised deep Kriging for single-image super-resolution | |
Lu et al. | Group-based single image super-resolution with online dictionary learning | |
Shang et al. | Super-resolution restoration of MMW image based on sparse representation method | |
Li et al. | Multi-scale cross-fusion for arbitrary scale image super resolution | |
Liang et al. | Multiband image fusion using total generalized variation regularization | |
Jeon et al. | PDE-guided reservoir computing for image denoising with small data | |
Zhu et al. | Multiframe blind restoration with image quality prior | |
Valli Bhasha et al. | Image super resolution model enabled by wavelet lifting with optimized deep convolutional neural network |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |