[發(fā)明專利]一種圖像處理方法及裝置、電子設(shè)備和存儲(chǔ)介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202110779002.5
• 申請(qǐng)日: 2021-07-09
• 公開（公告）號(hào): CN113378863A
• 公開（公告）日: 2021-09-10
• 發(fā)明（設(shè)計(jì)）人: 劉宇璽
• 申請(qǐng)（專利權(quán)）人: 上海商湯科技開發(fā)有限公司
• 主分類號(hào): G06K9/46
• 分類號(hào): G06K9/46;G06K9/62;G06F16/51;G06F17/16;G06F9/50;G06F9/54;G06T1/20;G06N3/04;G06N3/08
• 代理公司: 北京林達(dá)劉知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 11277
• 代理人: 劉新宇

1.一種圖像處理方法，其特征在于，包括：

對(duì)目標(biāo)圖像中的圖像特征進(jìn)行提取，得到用于表征圖像特征的特征圖；

通過多個(gè)第一線程計(jì)算特征圖和/或卷積核中的特征值的索引位置；

控制多個(gè)第二線程按照所述索引位置讀取特征圖和/或卷積核中的特征值，并利用讀取到的所述特征值進(jìn)行卷積處理，得到卷積特征，所述卷積特征用于表征對(duì)所述圖像特征的提取結(jié)果。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述利用讀取到的所述特征值進(jìn)行卷積處理，得到卷積特征，包括：

將讀取到的特征值在矩陣乘加運(yùn)算MMA指令的通道K維度進(jìn)行排列，進(jìn)行矩陣乘加運(yùn)算，得到卷積特征。

3.根據(jù)權(quán)利要求1或2所述的方法，其特征在于，所述通過多個(gè)第一線程計(jì)算特征圖和/卷積核中的特征值的索引位置，包括：

按照各線程的標(biāo)識(shí)ID的順序，控制各線程分別計(jì)算特征圖和/或卷積核中的特征值的索引位置。

4.根據(jù)權(quán)利要求1-3任一所述方法，其特征在于，在所述通過多個(gè)第一線程計(jì)算特征圖和/卷積核中的特征值的索引位置，所述方法還包括：

將所述索引位置的特征值加載到共享內(nèi)存SMem中，以供各線程使用。

5.根據(jù)權(quán)利要求1-4任一所述方法，其特征在于，所述控制多個(gè)第二線程按照所述索引位置讀取特征圖和/或卷積核中的特征值，包括：

控制所述多個(gè)第二線程將索引位置處的特征值讀入到各自的寄存器中；

所述利用讀取到的所述特征值進(jìn)行卷積處理，得到卷積特征，包括：

利用所述寄存器中的各個(gè)特征值進(jìn)行卷積處理，得到卷積特征。

6.根據(jù)權(quán)利要求1-5任一所述方法，其特征在于，在所述對(duì)目標(biāo)圖像中的圖像特征進(jìn)行提取后，所述方法還包括：

確定單個(gè)線程束Warp中能夠處理的特征塊的多種第一尺寸，并基于所述第一尺寸生成用于進(jìn)行特征塊劃分的圖形處理器核函數(shù)；其中，所述第一尺寸的最大值根據(jù)寄存器容量確定，所述第一尺寸的最小值為矩陣乘加運(yùn)算指令計(jì)算的最小矩陣單元的尺寸，所述多種第一尺寸的值為所述最小值的倍數(shù)。

7.根據(jù)權(quán)利要求1-6任一所述方法，其特征在于，在所述對(duì)目標(biāo)圖像中的圖像特征進(jìn)行提取后，所述方法還包括：

確定單個(gè)線程塊TB中能夠處理的特征塊的多種第二尺寸，并基于所述第二尺寸生成用于進(jìn)行特征塊劃分的圖形處理器核函數(shù)；其中，所述第二尺寸的最小值為單個(gè)線程束Warp能夠處理的特征塊的第一尺寸，所述第二尺寸為所述第一尺寸的倍數(shù)，所述第二尺寸的最大值根據(jù)共享內(nèi)存的容量和TB內(nèi)線程數(shù)上限確定。

8.一種圖像處理裝置，其特征在于，包括：

提取單元，用于對(duì)目標(biāo)圖像中的圖像特征進(jìn)行提取，得到用于表征圖像特征的特征圖；

索引計(jì)算單元，用于通過多個(gè)第一線程計(jì)算特征圖和/或卷積核中的特征值的索引位置；

卷積處理單元，用于控制多個(gè)第二線程按照所述索引位置讀取特征圖和/或卷積核中的特征值，并利用讀取到的所述特征值進(jìn)行卷積處理，得到卷積特征，所述卷積特征用于表征對(duì)所述圖像特征的提取結(jié)果。

9.一種電子設(shè)備，其特征在于，包括：

處理器；

用于存儲(chǔ)處理器可執(zhí)行指令的存儲(chǔ)器；

其中，所述處理器被配置為調(diào)用所述存儲(chǔ)器存儲(chǔ)的指令，以執(zhí)行權(quán)利要求1至7中任意一項(xiàng)所述的方法。

10.一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，其上存儲(chǔ)有計(jì)算機(jī)程序指令，其特征在于，所述計(jì)算機(jī)程序指令被處理器執(zhí)行時(shí)實(shí)現(xiàn)權(quán)利要求1至7中任意一項(xiàng)所述的方法。