公开(公告)号：US20240185573A1

公开(公告)日：2024-06-06

申请号：US18441229

申请日：2024-02-14

Applicant: HUAWEI TECHNOLOGIES CO., LTD.

Inventor： Han SHU ,  Jiahao WANG ,  Hanting CHEN ,  Wenshuo LI ,  Yunhe WANG

IPC: G06V10/77 ,  G06V10/764 ,  G06V10/80 ,  G06V10/82

CPC classification number: G06V10/7715 ,  G06V10/764 ,  G06V10/806 ,  G06V10/82

Abstract: This disclosure provides an image classification method and a related device thereof. The method includes the following operations: After obtaining a target image, a transformer network may perform linear transformation processing based on the target image to obtain a Q-feature, a K-feature, and a V-feature. The transformer network calculates a distance between the Q-feature and the K-feature to obtain an attention feature. Then, the transformer network performs fusion processing on the attention feature and the V-feature, and obtains a classification result of the target image based on a fused feature.