[發明專利]一種圖像檢測方法及裝置

在審

• 申請號: 201911024866.5
• 申請日: 2019-10-25
• 公開（公告）號: CN110781905A
• 公開（公告）日: 2020-02-11
• 發明（設計）人: 張水發;李巖
• 申請（專利權）人: 北京達佳互聯信息技術有限公司
• 主分類號: G06K9/46
• 分類號: G06K9/46;G06K9/62;G06K9/32;G06N3/04;G06N3/08
• 代理公司: 11413 北京柏杉松知識產權代理事務所(普通合伙)
• 代理人: 丁蕓;馬敬

本公開關于一種圖像檢測方法及裝置，通過獲取預先訓練的第二網絡，獲取第一分辨率的樣本圖像，其中，第二網絡為通過第二分辨率的樣本圖像訓練得到的圖像識別網絡，第一分辨率小于第二分辨率，對第一網絡進行訓練，得到訓練好的第一網絡，以通過訓練好的第一網絡進行圖像檢測，從而可以減小圖像檢測過程中網絡模型的計算量，提高計算速度。

技術領域

本公開涉及圖像管理的技術領域，尤其涉及一種圖像檢測方法及裝置。

背景技術

相關技術中，圖像對象檢測在人臉識別、醫學影像、智能視頻監控、機器人導航、基于內容的圖像檢索、基于圖像的繪制技術、圖像編輯和增強現實等領域都有廣泛的應用。圖像對象檢測時，通過利用圖像處理與模式識別等領域的理論和方法，從圖像中定位感興趣的圖形以及字符等目標，這需要準確地判斷每個目標的具體類別，并給出每個目標的邊界框等。

然而在圖像檢測中，由于往往存在較小的字或圖像。因此，檢測時就需要輸入較大的圖像，大大降低了檢測速度。而在很多需要實時處理的情況下，對圖像檢測的使用受到了很大的限制。特別是移動端，由于芯片處理能力較差，當輸入圖像分辨率太大時，大量的計算量不僅需要更多的處理時間，對芯片的功率需求也更大。因此，如何提高算法的速度，同時不降低準確率和召回率成為研究一個重點。

發明內容

本公開提供一種圖像檢測方法及裝置，以至少解決相關技術中檢測速度低的問題。本公開的技術方案如下：

根據本公開實施例的第一方面，提供一種圖像檢測方法，包括：

獲取預先訓練的第二網絡，獲取第一分辨率的樣本圖像，其中，第二網絡為通過第二分辨率的樣本圖像訓練得到的圖像識別網絡，第一分辨率小于第二分辨率；

通過待訓練的第一網絡的特征提取層，提取第一分辨率的樣本圖像的第一特征圖；

對第一特征圖進行上采樣，得到第二分辨率的上采樣特征圖；

將上采樣特征圖輸入第一網絡的預設區域生成層，并將第二網絡提取的特征信息作為第一網絡的預設區域生成層訓練過程的參考信息，獲取第一網絡的建議檢測框；

通過第一網絡的建議檢測框獲取上采樣特征圖的第一特征區域；

將第一特征區域輸入到第一網絡的感興趣區域池化層，并將第二網絡提取的特征信息作為第一網絡的預設區域生成層訓練過程的參考信息，獲取第一網絡針對樣本圖像的分類結果；

根據第一網絡針對樣本圖像的分類結果，調整第一網絡的參數繼續進行訓練，在滿足預設結束條件時，得到訓練好的第一網絡，以通過訓練好的第一網絡進行圖像檢測。

可選的，預先訓練第二網絡的步驟包括：

獲取第二分辨率的樣本圖像；

通過第二網絡的特征提取層，得到二分辨率的樣本圖像的第二特征圖；

將第二特征圖輸入第二網絡的預設區域生成層，得到第二網絡的建議檢測框；

通過第一網絡的建議檢測框獲取第二特征圖的第二特征區域；

將第二特征區域輸入到第二網絡的感興趣區域池化層，得到第二網絡針對樣本圖像的分類結果；

根據第二網絡針對樣本圖像的分類結果，調整第二網絡的參數繼續進行訓練，在滿足預設結束條件時，得到訓練好的第二網絡。

可選的，將上采樣特征圖輸入第一網絡的預設區域生成層，并將第二網絡提取的特征信息作為第一網絡的預設區域生成層訓練過程的參考信息，獲取第一網絡的建議檢測框，包括：

獲取第二網絡的預設區域生成層的第一分類標簽，其中，第一分類標簽是通過對第二特征圖進行第三類別回歸得到的分類標簽；