神經(jīng)網(wǎng)絡(luò)訓(xùn)練的方法、電子設(shè)備及計(jì)算機(jī)存儲(chǔ)介質(zhì)

基本信息

申請(qǐng)?zhí)?/td> CN202111307529.4 申請(qǐng)日 -
公開（公告）號(hào) CN114170438A 公開（公告）日 2022-03-11
申請(qǐng)公布號(hào) CN114170438A 申請(qǐng)公布日 2022-03-11
分類號(hào) G06V10/46（2022.01）I;G06V10/80（2022.01）I;G06V10/82（2022.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I;G06T7/55（2017.01）I 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 常青玲;崔巖;楊鑫;廖洹浩;王煜涵 申請(qǐng)（專利權(quán)）人 中德（珠海）人工智能研究院有限公司
代理機(jī)構(gòu) 廣州嘉權(quán)專利商標(biāo)事務(wù)所有限公司 代理人 梁國(guó)平
地址 529000廣東省江門市蓬江區(qū)東成村22號(hào)
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種神經(jīng)網(wǎng)絡(luò)訓(xùn)練的方法、電子設(shè)備及計(jì)算機(jī)存儲(chǔ)介質(zhì)，包括如下步驟：獲取預(yù)設(shè)的訓(xùn)練樣本圖；獲取預(yù)測(cè)特征圖；基于預(yù)設(shè)的邊界感知深度損失函數(shù)，以預(yù)設(shè)的訓(xùn)練樣本圖為輸入，預(yù)測(cè)特征圖為輸出，訓(xùn)練神經(jīng)網(wǎng)絡(luò)模型，得到訓(xùn)練后的神經(jīng)網(wǎng)絡(luò)模型。本發(fā)明通過(guò)設(shè)置邊界感知深度損失函數(shù)，能夠有效提高神經(jīng)網(wǎng)絡(luò)模型在訓(xùn)練過(guò)程中對(duì)邊界區(qū)域的關(guān)注度，保證邊界區(qū)域的深度和深度梯度具備良好的正確性，從而有效抑制平滑現(xiàn)象，避免出現(xiàn)邊界區(qū)域模糊的問(wèn)題，最終能夠有效提高邊界預(yù)測(cè)精度。