Domain adaptive image classification network training method, image classification method and device
A classification network and training method technology, applied in neural learning methods, biological neural network models, instruments, etc., can solve problems such as inability to solve the distribution of semantic information at the same level, and achieve the effect of improving classification accuracy
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0069] figure 1 is a flowchart of a training method for a domain-adapted image classification network according to an exemplary embodiment, as shown in figure 1 As shown, the method may include the following steps:
[0070] Step S11: obtaining several pairs of source domain images and target domain images, wherein the categories of each pair of source domain images and target domain images are the same;
[0071] Step S12: extracting a pair of cross-layer features of the source domain image and the target domain image;
[0072] Step S13: using the attention mechanism to calculate the similarity between the cross-layer features of the source domain image and the target domain image;
[0073] Step S14: Calculate the domain alignment generalization loss according to the multi-kernel maximum mean difference of the cross-layer feature and the similarity;
[0074] Step S15: Calculate the classification loss according to the cross-layer features of the source domain image and the t...
Embodiment 2
[0147] Figure 7 is a flow chart of an image classification method according to an exemplary embodiment, such as Figure 7 As shown, the method may include the following steps:
[0148] Step S51: acquiring the target domain image to be classified;
[0149] Specifically, for each target domain image, according to the above steps S21 and S22, first unify the size of the image, then unify the image size and perform normalization to obtain the target domain image x t .
[0150] Step S52: inputting the target domain image into a domain-adapted image classification network, wherein the domain-adapted image classification network is a network obtained by training according to the training method of the domain-adapted image classification network described in Embodiment 1;
[0151] Specifically, the encoding matrix x of the source domain image and the target domain image t Input the feature extractor F for feature extraction to obtain the second target domain feature f t . will ...
Embodiment 3
[0166] Correspondingly, the present application also provides an electronic device, comprising: one or more processors; a memory for storing one or more programs; when the one or more programs are executed by the one or more processors , so that the one or more processors implement the above-mentioned training method or image classification method of a domain-adapted image classification network.
PUM
Login to View More Abstract
Description
Claims
Application Information
Login to View More 


