[發(fā)明專利]目標(biāo)檢測模型、應(yīng)用方法及相關(guān)裝置

在審

• 申請?zhí)枺?/td>: 202010571484.0
• 申請日: 2020-06-19
• 公開（公告）號: CN111767947A
• 公開（公告）日: 2020-10-13
• 發(fā)明（設(shè)計）人: 尚太章
• 申請（專利權(quán)）人: OPPO廣東移動通信有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06N3/04;G06N3/08
• 代理公司: 廣州三環(huán)專利商標(biāo)代理有限公司 44202
• 代理人: 熊永強(qiáng)

本申請實施例公開了一種目標(biāo)檢測模型、應(yīng)用方法及相關(guān)裝置，包括：包括特征提取模塊、特征融合模塊和目標(biāo)檢測模塊，特征提取模塊連接特征融合模塊和目標(biāo)檢測模塊，特征融合模塊連接目標(biāo)檢測模塊；特征提取模塊，用于提取原始圖像的多種尺度的特征圖；特征融合模塊，用于對多種尺度的特征圖中至少三個特征圖進(jìn)行特征融合，至少三個特征圖包括尺度相同和不同的特征圖；目標(biāo)檢測模塊，用于根據(jù)多個特征圖中未進(jìn)行特征融合的特征圖和特征融合后的特征圖，得到原始圖像中待檢測目標(biāo)的目標(biāo)類別和目標(biāo)位置。本申請實施例在不過度增加模型復(fù)雜度的基礎(chǔ)上提高目標(biāo)檢測準(zhǔn)確度，針對輕量化的目標(biāo)檢測模型的小目標(biāo)檢測尤其適用。

技術(shù)領(lǐng)域

本申請涉及圖像處理技術(shù)領(lǐng)域，具體涉及一種目標(biāo)檢測模型、應(yīng)用方法及相關(guān)裝置。

背景技術(shù)

隨著人工智能技術(shù)的發(fā)展，目標(biāo)檢測在自動駕駛、行人檢測、車牌識別以及移動手機(jī)和AR眼鏡等智能終端得到廣泛應(yīng)用。特別是在智能終端中、現(xiàn)在已經(jīng)有大量智能算法集成進(jìn)去，用于進(jìn)一步提高終端智能。

近來隨著深度學(xué)習(xí)的出現(xiàn)以及爆炸式的發(fā)展，慢慢的也有很多研究人員開始關(guān)注于模型的輕量化，小型化，提高模型在終端設(shè)備的實際應(yīng)用價值。目標(biāo)檢測算法主要包括兩階段two-stage目標(biāo)檢測算法，Two-stage方法精度相對較高，但是模型時間復(fù)雜度較高，在終端設(shè)備部署比較難。

發(fā)明內(nèi)容

本申請實施例提供了一種目標(biāo)檢測模型、應(yīng)用方法及相關(guān)裝置，以期在不過度增加模型復(fù)雜度的基礎(chǔ)上提高目標(biāo)檢測準(zhǔn)確度，針對輕量化的目標(biāo)檢測模型的小目標(biāo)檢測尤其適用。

第一方面，本申請實施例提供一種目標(biāo)檢測模型，包括特征提取模塊、特征融合模塊和目標(biāo)檢測模塊，所述特征提取模塊連接所述特征融合模塊和所述目標(biāo)檢測模塊，所述特征融合模塊連接所述目標(biāo)檢測模塊；

所述特征提取模塊，用于提取原始圖像的多種尺度的特征圖；

所述特征融合模塊，用于對所述多種尺度的特征圖中至少三個特征圖進(jìn)行特征融合，所述至少三個特征圖包括尺度相同和不同的特征圖；

所述目標(biāo)檢測模塊，用于根據(jù)所述多個特征圖中未進(jìn)行特征融合的特征圖和特征融合后的特征圖，得到所述原始圖像中待檢測目標(biāo)的目標(biāo)類別和目標(biāo)位置。

第二方面，本申請實施例提供一種目標(biāo)檢測模型的訓(xùn)練方法，應(yīng)用于如第一方面所述的模型，所述方法包括：

設(shè)置所述目標(biāo)檢測模型的所述特征提取模塊、特征融合模塊和目標(biāo)檢測模塊的初始模塊參數(shù)；

在第一階段保持所述特征提取模塊的模塊參數(shù)不變，訓(xùn)練所述目標(biāo)檢測模型以更新所述特征融合模塊和所述目標(biāo)檢測模塊的模塊參數(shù)；

在第二階段訓(xùn)練所述目標(biāo)檢測模型以同步更新所述特征提取模塊、所述特征融合模塊和所述目標(biāo)檢測模塊的模塊參數(shù)，得到訓(xùn)練后的所述目標(biāo)檢測模型。

第三方面，本申請實施例提供一種目標(biāo)檢測模型的配置方法，應(yīng)用于如應(yīng)用于如第二方面所述的模型，所述方法包括：

服務(wù)器獲取訓(xùn)練好的目標(biāo)檢測模型；

所述服務(wù)器根據(jù)所述目標(biāo)檢測模型生成模型配置文件；

所述服務(wù)器向終端發(fā)送所述模型配置文件以配置所述目標(biāo)檢測模型。

第四方面，本申請實施例提供一種目標(biāo)檢測方法，應(yīng)用于配置有如第二方面所述的目標(biāo)檢測模型的終端，所述方法包括：

獲取原始圖像；

使用所述目標(biāo)檢測模型處理所述原始圖像，得到所述原始圖像中待檢測目標(biāo)的目標(biāo)類別和目標(biāo)位置。

第五方面，本申請實施例提供一種目標(biāo)檢測模型的訓(xùn)練裝置，應(yīng)用于如第一方面所述的模型，包括設(shè)置單元、第一訓(xùn)練單元和第二訓(xùn)練單元，