[發(fā)明專利]一種影像分類方法和系統(tǒng)

有效

• 申請?zhí)枺?/td>: 201610438777.5
• 申請日: 2016-06-17
• 公開（公告）號: CN107516061B
• 公開（公告）日: 2020-04-07
• 發(fā)明（設(shè)計(jì)）人: 石建萍;李聰
• 申請（專利權(quán)）人: 北京市商湯科技開發(fā)有限公司
• 主分類號: G06K9/00
• 分類號: G06K9/00;G06K9/62;G06N3/04
• 代理公司: 北京金闕華進(jìn)專利事務(wù)所(普通合伙) 11224
• 代理人: 陳建春

1.一種影像分類方法，其特征在于，所述方法包括：

將待分類的原始影像輸入訓(xùn)練好的卷積神經(jīng)網(wǎng)絡(luò)得到一融合特征；其中，所述卷積神經(jīng)網(wǎng)絡(luò)包括依次相連的對所述影像至少進(jìn)行卷積處理的多個(gè)特征提取處理級，所述原始影像為所述多個(gè)特征提取處理級中第一個(gè)特征提取處理級的輸入，部分特征提取處理級之間還分別連接一個(gè)步長不為1的中間池化層；所述卷積神經(jīng)網(wǎng)絡(luò)還包括分別對所述原始影像以及每個(gè)步長不為1的所述中間池化層輸出的輸出特征至少進(jìn)行卷積處理的多個(gè)特征保留處理級；所述卷積神經(jīng)網(wǎng)絡(luò)將所述多個(gè)特征保留處理級分別輸出的多個(gè)輸出特征以及所述多個(gè)特征提取處理級最終輸出的最終輸出特征的尺寸調(diào)節(jié)為與所述原始影像的尺寸一致，對尺寸調(diào)節(jié)后的所述多個(gè)輸出特征以及最終輸出特征進(jìn)行特征融合得到所述融合特征；

根據(jù)訓(xùn)練好的特征與類別之間的對應(yīng)關(guān)系，確定所述融合特征所對應(yīng)的所述原始影像所屬的類別。

2.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述多個(gè)特征提取處理級的至少部分特征提取處理級包括順次連接的卷積層、歸一化處理層和非線性變換處理層。

3.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述多個(gè)特征保留處理級中的每一個(gè)包括順次連接的卷積層、歸一化處理層、非線性變換處理層、丟棄層、卷積層、歸一化處理層、非線性變換處理層和卷積層。

4.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述對尺寸調(diào)節(jié)后的所述多個(gè)輸出特征以及最終輸出特征進(jìn)行特征融合得到所述融合特征包括：

對尺寸調(diào)節(jié)后的所述多個(gè)輸出特征的對應(yīng)位置的特征值相加得到第一影像特征；

將所述第一影像特征與尺寸調(diào)節(jié)后的最終輸出特征的對應(yīng)位置的特征值相加得到所述融合特征。

5.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述多個(gè)特征提取處理級包括7-14個(gè)特征提取處理級。

6.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述多個(gè)特征保留處理級包括2-7個(gè)特征保留處理級。

7.根據(jù)權(quán)利要求1所述的方法，其特征在于，所述中間池化層的卷積核大小為3×3。

8.根據(jù)權(quán)利要求1或2所述的方法，其特征在于，所述多個(gè)特征提取處理級中的卷積層的步長均為1。

9.一種影像分類系統(tǒng)，其特征在于，所述系統(tǒng)包括：

融合特征獲取模塊，用于將待分類的原始影像輸入訓(xùn)練好的卷積神經(jīng)網(wǎng)絡(luò)得到一融合特征；其中，所述卷積神經(jīng)網(wǎng)絡(luò)包括依次相連的對所述影像至少進(jìn)行卷積處理的多個(gè)特征提取處理級，所述原始影像為所述多個(gè)特征提取處理級中第一個(gè)特征提取處理級的輸入，部分特征提取處理級之間還分別連接一個(gè)步長不為1的中間池化層；所述卷積神經(jīng)網(wǎng)絡(luò)還包括分別對所述原始影像以及每個(gè)步長不為1的所述中間池化層輸出的輸出特征至少進(jìn)行卷積處理的多個(gè)特征保留處理級；所述卷積神經(jīng)網(wǎng)絡(luò)將所述多個(gè)特征保留處理級分別輸出的多個(gè)輸出特征以及所述多個(gè)特征提取處理級最終輸出的最終輸出特征的尺寸調(diào)節(jié)為與所述原始影像的尺寸一致，對尺寸調(diào)節(jié)后的所述多個(gè)輸出特征以及最終輸出特征進(jìn)行特征融合得到所述融合特征；

分類模塊，用于根據(jù)訓(xùn)練好的特征與類別之間的對應(yīng)關(guān)系，確定所述融合特征所對應(yīng)的所述原始影像所屬的類別。

10.根據(jù)權(quán)利要求9所述的系統(tǒng)，其特征在于，所述多個(gè)特征提取處理級的至少部分特征提取處理級包括順次連接的卷積層、歸一化處理層和非線性變換處理層。

11.根據(jù)權(quán)利要求9所述的系統(tǒng)，其特征在于，所述多個(gè)特征保留處理級中的每一個(gè)包括順次連接的卷積層、歸一化處理層、非線性變換處理層、丟棄層、卷積層、歸一化處理層、非線性變換處理層和卷積層。