一種圖像分類方法、裝置、終端及存儲(chǔ)介質(zhì)

基本信息

申請?zhí)?/td> CN202111583881.0 申請日 -
公開（公告）號 CN114219044A 公開（公告）日 2022-03-22
申請公布號 CN114219044A 申請公布日 2022-03-22
分類號 G06K9/62（2022.01）I;G06V10/40（2022.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I;G06V10/764（2022.01）I 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 雷柏英;熊唯;楊亮;于愷威;汪天富 申請（專利權(quán)）人 深圳大學(xué)
代理機(jī)構(gòu) 深圳市君勝知識產(chǎn)權(quán)代理事務(wù)所（普通合伙） 代理人 溫宏梅
地址 518060廣東省深圳市南山區(qū)南海大道3688號
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種圖像分類方法、裝置、終端及存儲(chǔ)介質(zhì)，方法通過獲取待分類圖像，將待分類圖像輸入目標(biāo)分類模型，其中，目標(biāo)分類模型包括卷積層、注意力層以及分類層；通過卷積層獲取待分類圖像的局部特征信息，得到第一特征圖；通過注意力層對第一特征圖進(jìn)行全局建模，得到第二特征圖；通過分類層對第二特征圖進(jìn)行圖像分類，得到待分類圖像對應(yīng)的圖像類別。本發(fā)明中的目標(biāo)分類模型可以捕捉待分類圖像的局部特征信息也可以進(jìn)行全局建模，因此可以精確地預(yù)測出待分類圖像對應(yīng)的圖像類別。解決了現(xiàn)有技術(shù)中深度卷積神經(jīng)網(wǎng)絡(luò)僅具備有捕捉局部上下文信息的能力，而不具備全局建模的能力，導(dǎo)致模型的分類性能不佳的問題。