[發明專利]基于空間通道的密度圖產生方法、裝置、電子終端及介質

有效

• 申請號: 201911045095.8
• 申請日: 2019-10-30
• 公開（公告）號: CN110837786B
• 公開（公告）日: 2022-07-08
• 發明（設計）人: 袁德勝;王作輝;張宏俊;王海濤;姚磊;楊進參;游浩泉;劉耀文;吳賀豐
• 申請（專利權）人: 匯納科技股份有限公司
• 主分類號: G06V20/52
• 分類號: G06V20/52;G06V10/44;G06N3/04
• 代理公司: 上海光華專利事務所(普通合伙) 31219
• 代理人: 龐紅芳

1.一種基于空間通道的密度圖產生方法，其特征在于，包括：

對輸入的原始圖像進行特征提取，以獲得初步特征圖；

根據預設的多個分割尺度對所述初步特征圖進行分割，且保留各分割塊中最大的值，以獲得對應所述多個分割尺度的中間特征圖；

將所述中間特征圖的任意兩點特征之間的位置關聯結果和所述中間特征圖的各通道之間的通道關聯結果進行融合，以獲得位置通道注意力特征圖；

將各所述分割尺度對應的位置通道注意力特征圖進行合并；所述合并包括：對所述位置通道注意力特征圖根據所述對應的分割尺度進行特征圖插值操作，以獲取與所述初步特征圖大小相同的多個輸出特征圖，且將所述的多個輸出特征圖在通道維度相接，獲得最終特征圖；

根據具有預設卷積尺度的卷積層對所述最終特征圖進行處理，以獲得與所述原始圖像大小相同的密度圖。

2.根據權利要求1所述的基于空間通道的密度圖產生方法，其特征在于，根據預設的多個不同的分割尺度對所述初步特征圖進行分割，且保留各分割塊中最大的值，以獲得對應所述多個分割尺度的中間特征圖。

3.根據權利要求1所述的基于空間通道的密度圖產生方法，其特征在于，將各所述分割尺度對應的位置通道注意力特征圖進行合并的步驟包括：

分別對各所述分割尺度對應的位置通道注意力特征圖進行插值，以使各所述分割尺度對應的位置通道注意力特征圖的大小相同；

且對進行插值后的各所述位置通道注意力特征圖逐次相加；

對逐次相加后獲得的位置通道注意力特征圖再次進行插值，以獲得與所述初步特征圖大小相同的最終特征圖。

4.根據權利要求1所述的基于空間通道的密度圖產生方法，其特征在于，根據具有預設卷積尺度的卷積層對所述最終特征圖進行處理的步驟包括：根據所述卷積層對所述最終特征圖進行降維處理，把具有多通道的所述最終特征圖將到一個通道，且對一個通道的所述最終特征圖執行圖像插值，以獲得與所述原始圖像大小相同的密度圖。

5.一種基于空間通道的密度圖產生裝置，其特征在于，包括：

特征提取模塊，用以對輸入的原始圖像進行特征提取，以獲得初步特征圖；

圖像分割模塊，用以根據預設的多個分割尺度對所述初步特征圖進行分割，且保留各分割塊中最大的值，以獲得對應所述多個分割尺度的中間特征圖；

注意力模塊，用以將所述中間特征圖的任意兩點特征之間的位置關聯結果和所述中間特征圖的各通道之間的通道關聯結果進行融合，以獲得位置通道注意力特征圖；

合并模塊，用以將各所述分割尺度對應的位置通道注意力特征圖進行合并；所述合并包括：對所述位置通道注意力特征圖根據所述對應的分割尺度進行特征圖插值操作，以獲取與所述初步特征圖大小相同的多個輸出特征圖，且用以將所述多個輸出特征圖在通道維度相接，獲得最終特征圖；

估計模塊，用以根據具有預設卷積尺度的卷積層對所述最終特征圖進行處理，以獲得與所述原始圖像大小相同的密度圖。

6.根據權利要求5所述的基于空間通道的密度圖產生裝置，其特征在于，所述圖像分割模塊還用以根據預設的多個不同的分割尺度對所述初步特征圖進行分割，且保留各分割塊中最大的值，以獲得對應所述多個分割尺度的中間特征圖；

所述合并模塊將各所述分割尺度對應的位置通道注意力特征圖進行合并的步驟包括：

分別對各所述分割尺度對應的位置通道注意力特征圖進行插值，以使各所述分割尺度對應的位置通道注意力特征圖的大小相同；

且對進行插值后的各所述位置通道注意力特征圖逐次相加；

對逐次相加后獲得的位置通道注意力特征圖再次進行插值，以獲得與所述初步特征圖大小相同的最終特征圖。

7.根據權利要求5所述的基于空間通道的密度圖產生裝置，其特征在于，所述估計模塊還用以根據所述卷積層對所述最終特征圖進行降維處理，把具有多通道的所述最終特征圖將到一個通道，且對一個通道的所述最終特征圖執行圖像插值，以獲得與所述原始圖像大小相同的密度圖。