[發(fā)明專利]一種神經(jīng)網(wǎng)絡(luò)訓(xùn)練方法及裝置

有效

• 申請(qǐng)?zhí)枺?/td>: 201710741903.9
• 申請(qǐng)日: 2017-08-25
• 公開(kāi)（公告）號(hào): CN107491790B
• 公開(kāi)（公告）日: 2021-04-02
• 發(fā)明（設(shè)計(jì)）人: 王乃巖;黃秀坤
• 申請(qǐng)（專利權(quán)）人: 北京圖森智途科技有限公司
• 主分類號(hào): G06K9/62
• 分類號(hào): G06K9/62;G06N3/08
• 代理公司: 暫無(wú)信息
• 代理人: 暫無(wú)信息

本發(fā)明公開(kāi)一種神經(jīng)網(wǎng)絡(luò)訓(xùn)練方法及裝置，以提高神經(jīng)網(wǎng)絡(luò)的健壯性和準(zhǔn)確性。方法包括：采用訓(xùn)練圖像集對(duì)神經(jīng)網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練，得到初級(jí)神經(jīng)網(wǎng)絡(luò)；將測(cè)試圖像集中的測(cè)試圖像輸入初級(jí)神經(jīng)網(wǎng)絡(luò)，輸出與測(cè)試圖像對(duì)應(yīng)的檢測(cè)結(jié)果；選取檢測(cè)結(jié)果有誤的測(cè)試圖像添加到訓(xùn)練圖像集中得到新的訓(xùn)練圖像集；采用新的訓(xùn)練圖像集對(duì)所述初級(jí)神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練。

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

本發(fā)明涉及計(jì)算機(jī)領(lǐng)域，特別涉及一種神經(jīng)網(wǎng)絡(luò)訓(xùn)練方法及裝置。

背景技術(shù)

目前，訓(xùn)練神經(jīng)網(wǎng)絡(luò)主要是基于預(yù)置的訓(xùn)練圖像集對(duì)神經(jīng)網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練得到目標(biāo)神經(jīng)網(wǎng)絡(luò)。訓(xùn)練圖像集中的圖像數(shù)量越多，基于該訓(xùn)練圖像集訓(xùn)練得到的目標(biāo)神經(jīng)網(wǎng)絡(luò)的準(zhǔn)確性越高。然而，在實(shí)際應(yīng)用中，訓(xùn)練圖像集的數(shù)量越多需要人工標(biāo)注的工作量越大，人力成本較高。因此，目前為平衡人力成本與神經(jīng)網(wǎng)絡(luò)準(zhǔn)確性，選取一定數(shù)量的圖像進(jìn)行標(biāo)注后構(gòu)成訓(xùn)練圖像集。

基于現(xiàn)有的訓(xùn)練圖像集，訓(xùn)練得到的神經(jīng)網(wǎng)絡(luò)僅適用于訓(xùn)練圖像集中出現(xiàn)的情況，對(duì)于訓(xùn)練圖像集中未出現(xiàn)的情況并不能很好的適用，準(zhǔn)確性還有待提高。

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

鑒于上述問(wèn)題，本發(fā)明提供一種神經(jīng)網(wǎng)絡(luò)訓(xùn)練方法及裝置，以提高神經(jīng)網(wǎng)絡(luò)的健壯性和準(zhǔn)確性。

本發(fā)明實(shí)施例，一方面，提供一種神經(jīng)網(wǎng)絡(luò)訓(xùn)練方法，該方法包括：

采用訓(xùn)練圖像集對(duì)神經(jīng)網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練，得到初級(jí)神經(jīng)網(wǎng)絡(luò)；

將測(cè)試圖像集中的測(cè)試圖像輸入初級(jí)神經(jīng)網(wǎng)絡(luò)，輸出與測(cè)試圖像對(duì)應(yīng)的檢測(cè)結(jié)果；

選取檢測(cè)結(jié)果有誤的測(cè)試圖像添加到訓(xùn)練圖像集中得到新的訓(xùn)練圖像集；

采用新的訓(xùn)練圖像集對(duì)所述初級(jí)神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練。

本發(fā)明實(shí)施例，還提供一種神經(jīng)網(wǎng)絡(luò)訓(xùn)練裝置，包括：

第一訓(xùn)練單元，用于采用訓(xùn)練圖像集對(duì)神經(jīng)網(wǎng)絡(luò)模型進(jìn)行訓(xùn)練，得到初級(jí)神經(jīng)網(wǎng)絡(luò)；

測(cè)試單元，用于將測(cè)試圖像集中的測(cè)試圖像輸入初級(jí)神經(jīng)網(wǎng)絡(luò)，輸出與測(cè)試圖像對(duì)應(yīng)的檢測(cè)結(jié)果；

更新單元，用于選取檢測(cè)結(jié)果有誤的測(cè)試圖像添加到訓(xùn)練圖像集中得到新的訓(xùn)練圖像集，并執(zhí)行第二訓(xùn)練單元；

第二訓(xùn)練單元，用于采用新的訓(xùn)練圖像集對(duì)所述初級(jí)神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練。

本發(fā)明實(shí)施例，在通過(guò)預(yù)置的訓(xùn)練圖像集訓(xùn)練神經(jīng)網(wǎng)絡(luò)模型得到初級(jí)神經(jīng)網(wǎng)絡(luò)之后，還通過(guò)測(cè)試圖像集對(duì)初級(jí)神經(jīng)網(wǎng)絡(luò)進(jìn)行測(cè)試，將檢測(cè)錯(cuò)誤的測(cè)試圖像(即正樣本)加入到訓(xùn)練圖像集中得到新的訓(xùn)練圖像集，再基于該新的訓(xùn)練圖像集對(duì)初級(jí)神經(jīng)網(wǎng)絡(luò)進(jìn)行訓(xùn)練。本發(fā)明實(shí)施例，自動(dòng)將測(cè)試圖像集中檢測(cè)有誤的測(cè)試圖像添加到訓(xùn)練圖像集中，由于檢測(cè)有誤的測(cè)試圖像具有反映初級(jí)神經(jīng)網(wǎng)絡(luò)檢測(cè)錯(cuò)誤的典型性，因此檢測(cè)有誤的測(cè)試圖像能夠形成很好的訓(xùn)練正樣本，將訓(xùn)練正樣本添加到訓(xùn)練圖像集中不僅能夠自動(dòng)擴(kuò)充訓(xùn)練圖像集，降低人力成本，而且基于新的訓(xùn)練圖像集訓(xùn)練得到的神經(jīng)網(wǎng)絡(luò)具有更強(qiáng)的健壯性和更低的錯(cuò)誤檢測(cè)率，提高神經(jīng)網(wǎng)絡(luò)的準(zhǔn)確性。

附圖說(shuō)明

附圖用來(lái)提供對(duì)本發(fā)明的進(jìn)一步理解，并且構(gòu)成說(shuō)明書(shū)的一部分，與本發(fā)明的實(shí)施例一起用于解釋本發(fā)明，并不構(gòu)成對(duì)本發(fā)明的限制。

圖1為本發(fā)明實(shí)施例中神經(jīng)網(wǎng)絡(luò)訓(xùn)練方法的流程圖之一；

圖2為本發(fā)明實(shí)施例中神經(jīng)網(wǎng)絡(luò)訓(xùn)練方法的流程圖之二；

圖3為本發(fā)明實(shí)施例中神經(jīng)網(wǎng)絡(luò)訓(xùn)練裝置的結(jié)構(gòu)示意圖之一；

圖4為本發(fā)明實(shí)施例中神經(jīng)網(wǎng)絡(luò)訓(xùn)練裝置的結(jié)構(gòu)示意圖之二。

具體實(shí)施方式