Image classification method and apparatus
This application relates to an image recognition technology in the field of computer vision in the field of artificial intelligence, and provides an image classification method and apparatus. The method includes: obtaining an input feature map of a to-be-processed image; performing convolution processing on the input feature map based on M convolution kernels of a neural network, to obtain a candidate output feature map of M channels, where M is a positive integer; performing matrix transformation on the M channels of the candidate output feature map based on N matrices, to obtain an output feature map of N channels, where a quantity of channels of each of the N matrices is less than M, N is greater than M, and N is a positive integer; and classify the to-be-processed image based on the output feature map, to obtain a classification result of the to-be-processed image.
1. An image classification method, comprising:
obtaining an input feature map of a to-be-processed image;
performing convolution processing on the input feature map based on M convolution kernels of a neural network, to obtain an intermediate feature map comprising M channels, wherein M is a positive integer;
performing convolution processing on the M channels of the intermediate feature map based on N matrices, to obtain an output feature map comprising N channels, wherein the N matrices represent N convolution kernels, the N matrices comprise M groups of convolution kernels, and the M groups of convolution kernels respectively correspond to the M channels of the intermediate feature map, and wherein a quantity of channels of each of the N matrices is less than M, N is greater than M, and N is a positive integer; and
classifying the to-be-processed image based on the output feature map, to obtain a classification result of the to-be-processed image.
2. The image classification method according to claim 1 , wherein
performing the convolution processing on the M channels of the intermediate feature map based on the N matrices, to obtain the output feature map comprising the N channels, comprises:
performing convolution on a corresponding channel of the M channels of the intermediate feature map based on each of the M groups of convolution kernels, to obtain the output feature map comprising the N channels.
3. The image classification method according to claim 2 , wherein performing the convolution on a corresponding channel of the M channels of the intermediate feature map based on each of the M groups of convolution kernels, to obtain the output feature map comprising the N channels, comprises:
performing, based on each of the M groups of convolution kernels, depthwise convolution on the corresponding channel of the M channels of the intermediate feature map, to obtain the output feature map comprising the N channels.
4. The image classification method according to claim 2 , wherein a convolution kernel in each of the M groups of convolution kernels is the same as a convolution kernel in another group of the M groups of convolution kernels.
5. The image classification method according to claim 1 , wherein classifying the to-be-processed image based on the output feature map, to obtain a classification result of the to-be-processed image, comprises:
combining the intermediate feature map and the output feature map to obtain a feature splicing feature map, wherein a quantity of channels of the feature splicing feature map is M+N; and
classifying the to-be-processed image based on the feature splicing feature map, to obtain the classification result of the to-be-processed image.
6. An image classification apparatus, comprising:
one or more processors; and
one or more memories coupled to the one or more processors and store programming instructions for execution by the one or more processors to:
obtain an input feature map of a to-be-processed image;
perform convolution processing on the input feature map based on M convolution kernels of a neural network, to obtain an intermediate feature map comprising M channels, wherein M is a positive integer;
perform convolution processing on the M channels of the intermediate feature map based on N matrices, to obtain an output feature map comprising N channels, wherein the N matrices represent N convolution kernels, the N matrices comprise M groups of convolution kernels, and the M groups of convolution kernels respectively correspond to the M channels of the intermediate feature map, and wherein a quantity of channels of each of the N matrices is less than M, N is greater than M, and N is a positive integer; and
classify the to-be-processed image based on the output feature map, to obtain a classification result of the to-be-processed image.
7. The image classification apparatus according to claim 6 , wherein
the one or more memories store programming instructions for execution by the one or more processors to perform convolution on a corresponding channel of the M channels of the intermediate feature map based on each of the M groups of convolution kernels, to obtain the output feature map comprising the N channels.
8. The image classification apparatus according to claim 7 , wherein the one or more memories store programming instructions for execution by the one or more processors to:
perform, based on each of the M groups of convolution kernels, depthwise convolution on the corresponding channel of the M channels of the intermediate feature map, to obtain the output feature map comprising the N channels.
9. The image classification apparatus according to claim 7 , wherein a convolution kernel in each of the M groups of convolution kernels is the same as a convolution kernel in another group of the M groups of convolution kernels.
10. The image classification apparatus according to claim 6 , wherein the one or more memories store programming instructions for execution by the one or more processors to:
combining the intermediate feature map and the output feature map to obtain a feature splicing feature map, wherein a quantity of channels of the feature splicing feature map is M+N; and
classify the to-be-processed image based on the feature splicing feature map, to obtain the classification result of the to-be-processed image.
11. A non-transitory computer-readable storage medium, wherein the non-transitory computer-readable storage medium stores one or more instructions that when executed by a device, cause the device to perform operations comprising:
obtaining an input feature map of a to-be-processed image;
performing convolution processing on the input feature map based on M convolution kernels of a neural network, to obtain an intermediate feature map comprising M channels, wherein M is a positive integer;
performing convolution processing on the M channels of the intermediate feature map based on N matrices, to obtain an output feature map comprising N channels, wherein the N matrices represent N convolution kernels, the N matrices comprise M groups of convolution kernels, and the M groups of convolution kernels respectively correspond to the M channels of the intermediate feature map, and wherein a quantity of channels of each of the N matrices is less than M, N is greater than M, and N is a positive integer; and
classifying the to-be-processed image based on the output feature map, to obtain a classification result of the to-be-processed image.
12. The non-transitory computer-readable storage medium according to claim 11 , wherein
performing the convolution processing on the M channels of the intermediate feature map based on the N matrices, to obtain the output feature map comprising the N channels, comprises:
performing convolution on a corresponding channel of the M channels of the intermediate feature map based on each of the M groups of convolution kernels, to obtain the output feature map comprising the N channels.
13. The non-transitory computer-readable storage medium according to claim 12 , wherein performing the convolution on a corresponding channel of the M channels of the intermediate feature map based on each of the M groups of convolution kernels, to obtain the output feature map comprising the N channels, comprises:
performing, based on each of the M groups of convolution kernels, depthwise convolution on the corresponding channel of the M channels of the intermediate feature map, to obtain the output feature map comprising the N channels.
14. The non-transitory computer-readable storage medium according to claim 12 , wherein a convolution kernel in each of the M groups of convolution kernels is the same as a convolution kernel in another group of the M groups of convolution kernels.
15. The non-transitory computer-readable storage medium according to claim 11 , wherein classifying the to-be-processed image based on the output feature map, to obtain a classification result of the to-be-processed image, comprises:
combining the intermediate feature map and the output feature map to obtain a feature splicing feature map, wherein a quantity of channels of the feature splicing feature map is M+N; and
classifying the to-be-processed image based on the feature splicing feature map, to obtain the classification result of the to-be-processed image.