[發(fā)明專利]目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法及系統(tǒng)及網(wǎng)絡(luò)及裝置及介質(zhì)

有效

• 申請(qǐng)?zhí)枺?/td>: 202010841756.4
• 申請(qǐng)日: 2020-08-20
• 公開（公告）號(hào): CN111709951B
• 公開（公告）日: 2020-11-13
• 發(fā)明（設(shè)計(jì)）人: 不公告發(fā)明人
• 申請(qǐng)（專利權(quán)）人: 成都數(shù)之聯(lián)科技有限公司
• 主分類號(hào): G06T7/00
• 分類號(hào): G06T7/00;G06K9/62;G06N3/08
• 代理公司: 成都云縱知識(shí)產(chǎn)權(quán)代理事務(wù)所(普通合伙) 51316
• 代理人: 熊曦;陳婉鵑

1.一種目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法，其特征在于，所述方法包括：

初始化分組比例K；

采集擁有標(biāo)注信息的訓(xùn)練數(shù)據(jù)，訓(xùn)練數(shù)據(jù)包括：原始輸入的圖像以及其對(duì)應(yīng)的標(biāo)注信息，標(biāo)注信息包括目標(biāo)框及掩碼；

將目標(biāo)掩碼以分組比例K劃分為partA和partB兩部分，目標(biāo)掩碼為一張圖像上所有的缺陷及各缺陷對(duì)應(yīng)的掩碼；

針對(duì)partA部分，結(jié)合目標(biāo)掩碼對(duì)應(yīng)的目標(biāo)掩碼信息，將原始圖像上對(duì)應(yīng)的目標(biāo)掩碼區(qū)域掩碼，得到history圖像；針對(duì)partB部分，保留partB部分在原始圖像上對(duì)應(yīng)的全部標(biāo)注信息，得到待輸入圖像；

將history圖像和待輸入圖像輸入待訓(xùn)練目標(biāo)檢測網(wǎng)絡(luò)，待輸入圖像經(jīng)過目標(biāo)檢測網(wǎng)第一層之后得到第一特征圖，history圖像經(jīng)過目標(biāo)檢測網(wǎng)第一層之后得到第二特征圖，將第一特征圖與第二特征圖相加后輸入目標(biāo)檢測網(wǎng)后續(xù)層進(jìn)行相關(guān)計(jì)算訓(xùn)練目標(biāo)檢測網(wǎng)絡(luò)。

2.根據(jù)權(quán)利要求1所述的目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法，其特征在于，所述針對(duì)partA部分，結(jié)合目標(biāo)掩碼對(duì)應(yīng)的目標(biāo)掩碼信息，將原始圖像上對(duì)應(yīng)的目標(biāo)掩碼區(qū)域掩碼，具體包括：

在原始圖像中將partA部分目標(biāo)掩碼所覆蓋的像素置為0，并在原始groundtruth的回歸框和目標(biāo)掩碼中去掉partA部分目標(biāo)缺陷對(duì)應(yīng)的標(biāo)注信息，原始groundtruth為圖像上缺陷的標(biāo)注信息。

3.根據(jù)權(quán)利要求1所述的目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法，其特征在于，將第一特征圖與第二特征圖相加具體為：將第一特征圖與第二特征圖矩陣按位相加，即兩個(gè)特征圖矩陣上對(duì)應(yīng)坐標(biāo)的值進(jìn)行相加。

4.根據(jù)權(quán)利要求1所述的目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法，其特征在于，所述方法還包括測試和驗(yàn)證訓(xùn)練后的目標(biāo)檢測網(wǎng)絡(luò)。

5.一種目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練系統(tǒng)，其特征在于，所述系統(tǒng)包括：

初始化單元，用于初始化分組比例K；

標(biāo)注單元，用于采集擁有標(biāo)注信息的訓(xùn)練數(shù)據(jù)，對(duì)訓(xùn)練數(shù)據(jù)標(biāo)注信息回歸框，訓(xùn)練數(shù)據(jù)包括：原始輸入的圖像以及其對(duì)應(yīng)的標(biāo)注信息，標(biāo)注信息包括目標(biāo)框及掩碼；

分組單元，用于將目標(biāo)掩碼以分組比例K劃分為partA和partB兩部分，目標(biāo)掩碼為一張圖像上所有的缺陷及各缺陷對(duì)應(yīng)的掩碼；

掩碼單元，用于針對(duì)partA部分，結(jié)合目標(biāo)掩碼對(duì)應(yīng)的目標(biāo)掩碼信息，將原始圖像上對(duì)應(yīng)的目標(biāo)掩碼區(qū)域掩碼，得到history圖像；針對(duì)partB部分，保留partB部分在原始圖像上對(duì)應(yīng)的全部標(biāo)注信息，得到待輸入圖像；

訓(xùn)練單元，用于將history圖像和待輸入圖像輸入待訓(xùn)練目標(biāo)檢測網(wǎng)絡(luò)，待輸入圖像經(jīng)過目標(biāo)檢測網(wǎng)第一層之后得到第一特征圖，history圖像經(jīng)過目標(biāo)檢測網(wǎng)第一層之后得到第二特征圖，將第一特征圖與第二特征圖相加后輸入目標(biāo)檢測網(wǎng)后續(xù)層進(jìn)行相關(guān)計(jì)算訓(xùn)練目標(biāo)檢測網(wǎng)絡(luò)。

6.一種目標(biāo)檢測網(wǎng)絡(luò)，其特征在于，所述目標(biāo)檢測網(wǎng)絡(luò)用于對(duì)預(yù)設(shè)目標(biāo)進(jìn)行檢測，所述目標(biāo)檢測網(wǎng)絡(luò)采用權(quán)利要求1-4中任意一種所述的目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法訓(xùn)練獲得。

7.一種目標(biāo)檢測系統(tǒng)，其特征在于，所述系統(tǒng)包括：

采集器，用于采集待檢測圖像；

處理器，所述處理器用于運(yùn)行目標(biāo)檢測網(wǎng)絡(luò)或模型，所述目標(biāo)檢測網(wǎng)絡(luò)或模型用于對(duì)待檢測圖像進(jìn)行目標(biāo)檢測，并輸出目標(biāo)檢測結(jié)果，所述目標(biāo)檢測網(wǎng)絡(luò)或模型采用權(quán)利要求1-4中任意一種所述的目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法訓(xùn)練獲得。

8.一種目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練裝置，包括存儲(chǔ)器、處理器以及存儲(chǔ)在所述存儲(chǔ)器中并可在所述處理器上運(yùn)行的計(jì)算機(jī)程序，其特征在于，所述處理器執(zhí)行所述計(jì)算機(jī)程序時(shí)實(shí)現(xiàn)如權(quán)利要求1-4中任意一個(gè)所述的目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法的步驟。

9.一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)，所述計(jì)算機(jī)可讀存儲(chǔ)介質(zhì)存儲(chǔ)有計(jì)算機(jī)程序，其特征在于，所述計(jì)算機(jī)程序被處理器執(zhí)行時(shí)實(shí)現(xiàn)如權(quán)利要求1-4中任意一個(gè)所述的目標(biāo)檢測網(wǎng)絡(luò)訓(xùn)練方法的步驟。