-
公开(公告)号:CN115712828A
公开(公告)日:2023-02-24
申请号:CN202110950638.1
申请日:2021-08-18
Applicant: 华为技术有限公司
IPC: G06F18/214 , G06F18/213 , G06F18/241
Abstract: 本申请提供一种图像分类方法及其相关设备,该方法包括:在获取目标图像后,transformer网络可基于目标图像进行线性变换处理,得到Q特征、K特征和V特征。接着,transformer网络计算Q特征和K特征之间的距离,从而得到注意力特征。然后,transformer网络将注意力特征和V特征进行融合处理,并基于融合后的特征获取目标图像的分类结果。前述过程中,由于Q特征和K特征之间的距离的计算主要涉及加法运算,故transformer网络获取注意力特征的操作,不再包含大量的乘法运算,而是被计算开销较小的加法运算所代替,可使得transformer网络广泛应用在算力有限的终端设备上。