基于梯度維持的圖像分類方法、存儲介質(zhì)及電子設(shè)備

基本信息

申請?zhí)?/td> CN202011377685.3 申請日 -
公開（公告）號 CN112348118A 公開（公告）日 2021-02-09
申請公布號 CN112348118A 申請公布日 2021-02-09
分類號 G06K9/62(2006.01)I;G06N3/04(2006.01)I 分類 計算；推算；計數(shù);
發(fā)明人 何良華;任強;施小春;李鵬飛;羅濤 申請（專利權(quán)）人 華平信息技術(shù)股份有限公司
代理機構(gòu) 上海光華專利事務(wù)所（普通合伙） 代理人 徐秋平
地址 200433上海市楊浦區(qū)國權(quán)北路1688弄69號
法律狀態(tài) -

摘要

摘要 本發(fā)明提供一種基于梯度維持的圖像分類方法、存儲介質(zhì)及電子設(shè)備，所述基于梯度維持的圖像分類方法包括：獲取待訓(xùn)練的圖像集；將所述圖像集劃分為訓(xùn)練集和測試集；利用所述訓(xùn)練集訓(xùn)練分類膠囊網(wǎng)絡(luò)模型，且在訓(xùn)練過程中通過圖像預(yù)測膠囊的路由梯度系數(shù)維持膠囊層疊加時的梯度穩(wěn)定；將所述測試集輸入所述分類膠囊網(wǎng)絡(luò)模型中進行圖像分類的性能測試。本發(fā)明在圖像分類所用的膠囊網(wǎng)絡(luò)模型中通過路由梯度系數(shù)維持梯度穩(wěn)定。不僅在某種程度上抑制了梯度的消失，而且還可以適當(dāng)?shù)目梢苑糯筇荻炔⑻荻雀交財U展到模型前面的網(wǎng)絡(luò)層，提高了膠囊網(wǎng)絡(luò)模型的圖像分類性能。??