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

在審

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

本公開提供一種貨物圖像局部區域處理方法、裝置、設備及存儲介質，涉及圖像處理技術領域。該方法包括：獲取訓練圖像和標簽；通過特征提取網絡對從所述訓練圖像中提取的特征進行注意力增強處理獲得注意力增強特征圖；將所述注意力增強特征圖通過區域建議網絡進行處理獲得區域建議圖；基于所述注意力增強特征圖和所述區域建議圖通過區域分類網絡獲得區域分類圖；基于所述區域分類圖和所述標簽獲得損失函數；以及根據所述損失函數訓練所述特征提取網絡、所述區域建議網絡和所述區域分類網絡以處理貨物圖像局部區域。該方法一定程度上提高了提取貨物圖像局部區域的準確性。

技術領域

本公開涉及圖像處理技術領域，具體而言，涉及一種貨物圖像局部區域處理方法、裝置、設備及可讀存儲介質。

背景技術

在海關查驗過程中不僅需要能夠檢出整箱大宗嫌疑貨物，檢出局部嫌疑貨物也是非常重要的海關查驗工作。通常采用X射線設備探測海關進出口貨物獲得貨物的輻射圖像，以實現不開箱貨物查驗，準確地提取輻射圖像中的貨物局部區域是局部嫌疑貨物檢出的重要前提。但是，由于海關吞吐的貨物形態復雜，裝載方式多樣，并且受X射線成像角度差異的影響，準確地劃分輻射圖像的局部貨物區域非常困難。

如上所述，如何提供較準確地劃分局部貨物區域的貨物圖像處理方法成為亟待解決的問題。

在所述背景技術部分公開的上述信息僅用于加強對本公開的背景的理解，因此它可以包括不構成對本領域普通技術人員已知的現有技術的信息。

發明內容

本公開的目的在于提供一種貨物圖像局部區域處理方法、裝置、設備及可讀存儲介質，至少在一定程度上克服相關技術難以準確地劃分輻射圖像的局部貨物區域的問題。

本公開的其他特性和優點將通過下面的詳細描述變得顯然，或部分地通過本公開的實踐而習得。

根據本公開的一方面，提供一種貨物圖像局部區域處理方法，包括：獲取訓練圖像和標簽；通過特征提取網絡對從所述訓練圖像中提取的特征進行注意力增強處理獲得注意力增強特征圖；將所述注意力增強特征圖通過區域建議網絡進行處理獲得區域建議圖；基于所述注意力增強特征圖和所述區域建議圖通過區域分類網絡獲得區域分類圖；基于所述區域分類圖和所述標簽獲得損失函數；以及根據所述損失函數訓練所述特征提取網絡、所述區域建議網絡和所述區域分類網絡以處理貨物圖像局部區域。

根據本公開的一實施例，所述特征提取網絡包括卷積層和全連接層；所述通過特征提取網絡對從所述訓練圖像中提取的特征進行注意力增強處理獲得注意力增強特征圖包括：通過所述卷積層提取所述訓練圖像的多個通道的特征；將所述多個通道的特征通過所述全連接層進行通道加權處理獲得所述多個通道中各個通道的權重；以及將所述多個通道的特征與所述各個通道的權重進行融合獲得通道加權特征圖；對所述通道加權特征圖進行分組增強處理獲得所述注意力增強特征圖。

根據本公開的一實施例，所述對所述通道加權特征圖進行分組增強處理獲得所述注意力增強特征圖包括：將所述通道加權特征圖中的特征矩陣分成多組獲得多個特征組；基于所述多個特征組獲得增強系數矩陣；以及將所述通道加權特征圖中的特征矩陣與所述增強系數矩陣相乘獲得所述注意力增強特征圖。

根據本公開的一實施例，所述基于所述多個特征組獲得增強系數矩陣包括：對所述多個特征組中的各個特征組進行全局平均池化處理獲得所述各個特征組的組全局特征；將所述各個特征組的組全局特征分別與所述各個特征組中的特征進行點乘獲得相似性矩陣；以及對所述相似性矩陣進行歸一化處理后獲得所述增強系數矩陣。