基于通道注意力的圖像語義分割方法及裝置

基本信息

申請?zhí)?/td> CN202110837049.2 申請日 -
公開（公告）號 CN113361537A 公開（公告）日 2021-09-07
申請公布號 CN113361537A 申請公布日 2021-09-07
分類號 G06K9/34;G06K9/46;G06N3/04;G06N3/08 分類 計算；推算；計數(shù);
發(fā)明人 郭俊波;郭筱鳳;靳國慶;馬凌峰;謝洪濤;張勇東 申請（專利權）人 人民網股份有限公司
代理機構 北京市浩天知識產權代理事務所（普通合伙） 代理人 宋菲
地址 100026 北京市西城區(qū)新街口外大街28號B座234號
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于通道注意力的圖像語義分割方法及裝置，該方法包括：對原始圖像進行預處理，得到待分割的輸入圖；將輸入圖輸入至圖像語義分割模型的特征提取網絡中，利用特征提取網絡對輸入圖進行特征提取，得到特征圖；其中，特征提取網絡的各個階段的各個瓶頸模塊中插入有通道注意力模塊，用于將輸入特征經過水平池化和豎直池化，得到兩個矩陣；從通道角度使用1維卷積并行處理兩個矩陣，并進行歸一化處理后取平均值，得到通道權重；將特征圖輸入至圖像語義分割模型的解碼器中，利用解碼器對特征圖進行處理，得到預測分割圖。本發(fā)明是通過通道注意力機制改進編碼器的性能，能夠很方便地移植到各種分割網絡中，能夠有明顯的性能提升。