基于梯度維持的圖像分類方法、存儲(chǔ)介質(zhì)及電子設(shè)備
基本信息
| 申請(qǐng)?zhí)?/td> | CN202011377685.3 | 申請(qǐng)日 | - |
| 公開（公告）號(hào) | CN112348118A | 公開（公告）日 | 2021-02-09 |
| 申請(qǐng)公布號(hào) | CN112348118A | 申請(qǐng)公布日 | 2021-02-09 |
| 分類號(hào) | G06K9/62(2006.01)I;G06N3/04(2006.01)I | 分類 | 計(jì)算；推算；計(jì)數(shù); |
| 發(fā)明人 | 何良華;任強(qiáng);施小春;李鵬飛;羅濤 | 申請(qǐng)（專利權(quán)）人 | 華平信息技術(shù)股份有限公司 |
| 代理機(jī)構(gòu) | 上海光華專利事務(wù)所（普通合伙） | 代理人 | 徐秋平 |
| 地址 | 200433上海市楊浦區(qū)國(guó)權(quán)北路1688弄69號(hào) | ||
| 法律狀態(tài) | - | ||
摘要
| 摘要 | 本發(fā)明提供一種基于梯度維持的圖像分類方法、存儲(chǔ)介質(zhì)及電子設(shè)備，所述基于梯度維持的圖像分類方法包括：獲取待訓(xùn)練的圖像集；將所述圖像集劃分為訓(xùn)練集和測(cè)試集；利用所述訓(xùn)練集訓(xùn)練分類膠囊網(wǎng)絡(luò)模型，且在訓(xùn)練過(guò)程中通過(guò)圖像預(yù)測(cè)膠囊的路由梯度系數(shù)維持膠囊層疊加時(shí)的梯度穩(wěn)定；將所述測(cè)試集輸入所述分類膠囊網(wǎng)絡(luò)模型中進(jìn)行圖像分類的性能測(cè)試。本發(fā)明在圖像分類所用的膠囊網(wǎng)絡(luò)模型中通過(guò)路由梯度系數(shù)維持梯度穩(wěn)定。不僅在某種程度上抑制了梯度的消失，而且還可以適當(dāng)?shù)目梢苑糯筇荻炔⑻荻雀交財(cái)U(kuò)展到模型前面的網(wǎng)絡(luò)層，提高了膠囊網(wǎng)絡(luò)模型的圖像分類性能。?? |
