[發(fā)明專(zhuān)利]模型生成方法、目標(biāo)檢測(cè)方法、裝置、電子設(shè)備及介質(zhì)

在審

• 申請(qǐng)?zhí)枺?/td>: 202011391459.0
• 申請(qǐng)日: 2020-12-03
• 公開(kāi)（公告）號(hào): CN112257815A
• 公開(kāi)（公告）日: 2021-01-22
• 發(fā)明（設(shè)計(jì)）人: 程鵬;何凌霄;廖星宇;王林芳;劉武;梅濤
• 申請(qǐng)（專(zhuān)利權(quán)）人: 北京沃東天駿信息技術(shù)有限公司;北京京東世紀(jì)貿(mào)易有限公司
• 主分類(lèi)號(hào): G06K9/62
• 分類(lèi)號(hào): G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京品源專(zhuān)利代理有限公司 11332
• 代理人: 孟金喆

1.一種模型生成方法，其特征在于，包括：

基于多組包括第一樣本圖像和所述第一樣本圖像中已知目標(biāo)的樣本標(biāo)注結(jié)果的教師訓(xùn)練樣本對(duì)原始檢測(cè)模型進(jìn)行訓(xùn)練，得到教師網(wǎng)絡(luò)；

將所述第一樣本圖像和在將所述第一樣本圖像輸入至所述教師網(wǎng)絡(luò)后得到的第一檢測(cè)結(jié)果作為第一訓(xùn)練樣本，并且將第二樣本圖像和在將所述第二樣本圖像輸入至所述教師網(wǎng)絡(luò)后得到的第二檢測(cè)結(jié)果作為第二訓(xùn)練樣本；

基于多組所述第一訓(xùn)練樣本和多組所述第二訓(xùn)練樣本對(duì)與所述教師網(wǎng)絡(luò)具有同一網(wǎng)絡(luò)類(lèi)型的學(xué)生網(wǎng)絡(luò)進(jìn)行訓(xùn)練，生成目標(biāo)檢測(cè)模型。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述與所述教師網(wǎng)絡(luò)具有同一網(wǎng)絡(luò)類(lèi)型的學(xué)生網(wǎng)絡(luò)包括目標(biāo)預(yù)測(cè)模塊和損失計(jì)算模塊；

針對(duì)每組所述第一訓(xùn)練樣本，所述基于多組所述第一訓(xùn)練樣本對(duì)與所述教師網(wǎng)絡(luò)具有同一網(wǎng)絡(luò)類(lèi)型的學(xué)生網(wǎng)絡(luò)進(jìn)行訓(xùn)練，包括：

將所述第一樣本圖像輸入至所述目標(biāo)預(yù)測(cè)模塊中，得到第一預(yù)測(cè)結(jié)果；

將所述第一檢測(cè)結(jié)果和所述第一預(yù)測(cè)結(jié)果輸入至所述損失計(jì)算模塊中，并根據(jù)所述損失計(jì)算模塊的輸出結(jié)果調(diào)節(jié)所述目標(biāo)預(yù)測(cè)模塊中的網(wǎng)絡(luò)參數(shù)。

3.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述第一檢測(cè)結(jié)果包括中心點(diǎn)檢測(cè)特征圖、寬高檢測(cè)特征圖和/或中心點(diǎn)偏移量檢測(cè)特征圖；相應(yīng)的，所述第一預(yù)測(cè)結(jié)果包括中心點(diǎn)預(yù)測(cè)特征圖、寬高預(yù)測(cè)特征圖和/或中心點(diǎn)偏移量預(yù)測(cè)特征圖；相應(yīng)的，所述損失計(jì)算模塊包括中心點(diǎn)損失計(jì)算模塊、寬高損失計(jì)算模塊和/或中心點(diǎn)偏移量損失計(jì)算模塊；

所述將所述第一檢測(cè)結(jié)果和所述第一預(yù)測(cè)結(jié)果輸入至所述損失計(jì)算模塊中，并根據(jù)所述損失計(jì)算模塊的輸出結(jié)果調(diào)節(jié)所述目標(biāo)預(yù)測(cè)模塊中的網(wǎng)絡(luò)參數(shù)包括：

將所述中心點(diǎn)檢測(cè)特征圖和所述中心點(diǎn)預(yù)測(cè)特征圖輸入至所述中心點(diǎn)損失計(jì)算模塊，將所述寬高檢測(cè)特征圖和所述寬高預(yù)測(cè)特征圖輸入至所述寬高損失計(jì)算模塊，和/或?qū)⑺鲋行狞c(diǎn)偏移量檢測(cè)特征圖和所述中心點(diǎn)偏移量預(yù)測(cè)特征圖輸入至所述中心點(diǎn)偏移量損失計(jì)算模塊；

根據(jù)所述中心點(diǎn)損失計(jì)算模塊、所述寬高損失計(jì)算模塊、和/或所述中心點(diǎn)偏移量損失計(jì)算模塊的輸出結(jié)果，確定損失計(jì)算結(jié)果；

根據(jù)所述損失計(jì)算結(jié)果調(diào)節(jié)所述目標(biāo)預(yù)測(cè)模塊中的網(wǎng)絡(luò)參數(shù)。

4.根據(jù)權(quán)利要求3所述的方法，其特征在于，所述中心點(diǎn)損失計(jì)算模塊包括用于對(duì)所述中心點(diǎn)檢測(cè)特征圖和所述中心點(diǎn)預(yù)測(cè)特征圖進(jìn)行激活處理的中心點(diǎn)特征圖激活單元、以及用于對(duì)激活處理后的所述中心點(diǎn)檢測(cè)特征圖和所述中心點(diǎn)預(yù)測(cè)特征圖間的損失數(shù)值進(jìn)行計(jì)算的中心點(diǎn)損失計(jì)算單元；和/或，

所述寬高損失計(jì)算模塊，用于根據(jù)類(lèi)別概率特征圖對(duì)所述寬高檢測(cè)特征圖和所述寬高預(yù)測(cè)特征圖間的損失數(shù)值進(jìn)行計(jì)算，其中，所述類(lèi)別概率特征圖是所述中心點(diǎn)檢測(cè)特征圖在類(lèi)別概率維度上取最大值時(shí)生成的特征圖；和/或，

所述中心點(diǎn)偏移量損失計(jì)算模塊，用于根據(jù)所述類(lèi)別概率特征圖對(duì)所述中心點(diǎn)偏移量檢測(cè)特征圖和所述中心點(diǎn)偏移量預(yù)測(cè)特征圖間的損失數(shù)值進(jìn)行計(jì)算。

5.根據(jù)權(quán)利要求2所述的方法，其特征在于，所述根據(jù)所述損失計(jì)算模塊的輸出結(jié)果調(diào)節(jié)所述目標(biāo)預(yù)測(cè)模塊中的網(wǎng)絡(luò)參數(shù)，包括：

根據(jù)預(yù)先輸出結(jié)果和損失數(shù)值范圍確定損失系數(shù)，所述預(yù)先輸出結(jié)果包括預(yù)先對(duì)所述學(xué)生網(wǎng)絡(luò)進(jìn)行預(yù)設(shè)次數(shù)的迭代訓(xùn)練后所述損失計(jì)算模塊的輸出結(jié)果；

根據(jù)所述損失系數(shù)、以及將所述第一檢測(cè)結(jié)果和所述第一預(yù)測(cè)結(jié)果輸入至所述損失計(jì)算模塊后所述損失計(jì)算模塊的輸出結(jié)果，調(diào)節(jié)所述目標(biāo)預(yù)測(cè)模塊中的網(wǎng)絡(luò)參數(shù)。

6.根據(jù)權(quán)利要求5所述的方法，其特征在于，所述預(yù)先輸出結(jié)果通過(guò)如下步驟預(yù)先確定：

在所述基于多組所述第一訓(xùn)練樣本和多組所述第二訓(xùn)練樣本對(duì)與所述教師網(wǎng)絡(luò)具有同一網(wǎng)絡(luò)類(lèi)型的學(xué)生網(wǎng)絡(luò)進(jìn)行訓(xùn)練之前，基于多組所述第一訓(xùn)練樣本和多組所述第二訓(xùn)練樣本對(duì)所述學(xué)生網(wǎng)絡(luò)進(jìn)行所述預(yù)設(shè)次數(shù)的迭代訓(xùn)練，并將訓(xùn)練結(jié)束時(shí)所述損失計(jì)算模塊的輸出結(jié)果作為所述預(yù)先輸出結(jié)果。