[發明專利]貨物圖像局部區域處理方法、裝置、設備及存儲介質

在審

• 申請號: 202010512064.5
• 申請日: 2020-06-08
• 公開（公告）號: CN113837172A
• 公開（公告）日: 2021-12-24
• 發明（設計）人: 陳志強;李元景;張麗;邢宇翔;戴詩語;丁保闊;郭全占
• 申請（專利權）人: 同方威視科技江蘇有限公司;同方威視技術股份有限公司
• 主分類號: G06K9/32
• 分類號: G06K9/32;G06K9/62;G06N3/04
• 代理公司: 北京律智知識產權代理有限公司 11438
• 代理人: 孫寶海;袁禮君

1.一種貨物圖像局部區域處理方法，其特征在于，包括：

獲取訓練圖像和標簽；

通過特征提取網絡對從所述訓練圖像中提取的特征進行注意力增強處理獲得注意力增強特征圖；

將所述注意力增強特征圖通過區域建議網絡進行處理獲得區域建議圖；

基于所述注意力增強特征圖和所述區域建議圖通過區域分類網絡獲得區域分類圖；

基于所述區域分類圖和所述標簽獲得損失函數；以及

根據所述損失函數訓練所述特征提取網絡、所述區域建議網絡和所述區域分類網絡以處理貨物圖像局部區域。

2.根據權利要求1所述的方法，其特征在于，所述特征提取網絡包括卷積層和全連接層；

所述通過特征提取網絡對從所述訓練圖像中提取的特征進行注意力增強處理獲得注意力增強特征圖包括：

通過所述卷積層提取所述訓練圖像的多個通道的特征；

將所述多個通道的特征通過所述全連接層進行通道加權處理獲得所述多個通道中各個通道的權重；以及

將所述多個通道的特征與所述各個通道的權重進行融合獲得通道加權特征圖；

對所述通道加權特征圖進行分組增強處理獲得所述注意力增強特征圖。

3.根據權利要求2所述的方法，其特征在于，所述對所述通道加權特征圖進行分組增強處理獲得所述注意力增強特征圖包括：

將所述通道加權特征圖中的特征矩陣分成多組獲得多個特征組；

基于所述多個特征組獲得增強系數矩陣；以及

將所述通道加權特征圖中的特征矩陣與所述增強系數矩陣相乘獲得所述注意力增強特征圖。

4.根據權利要求3所述的方法，其特征在于，所述基于所述多個特征組獲得增強系數矩陣包括：

對所述多個特征組中的各個特征組進行全局平均池化處理獲得所述各個特征組的組全局特征；

將所述各個特征組的組全局特征分別與所述各個特征組中的特征進行點乘獲得相似性矩陣；以及

對所述相似性矩陣進行歸一化處理后獲得所述增強系數矩陣。

5.根據權利要求1所述的方法，其特征在于，所述區域分類網絡包括全連接區域分類網絡和全卷積區域分類網絡；以及

所述區域分類圖包括感興趣區域和所述感興趣區域的類別；

所述基于所述注意力增強特征圖和所述區域建議圖通過區域分類網絡獲得區域分類圖包括：

根據所述注意力增強特征圖提取所述區域建議圖中區域建議框的特征獲得注意力增強區域特征圖；

將所述注意力增強區域特征圖通過所述全卷積區域分類網絡后進行定位處理獲得所述感興趣區域；以及

將所述注意力增強區域特征圖通過所述全連接區域分類網絡后進行分類處理獲得所述感興趣區域的類別；

所述基于所述區域分類圖和所述標簽獲得損失函數包括：

基于所述感興趣區域和所述標簽獲得區域損失項；

基于所述感興趣區域的類別和所述標簽獲得分類損失項；以及

根據所述區域損失項和所述分類損失項獲得所述損失函數。