[發(fā)明專利]基于空間通道的密度圖產(chǎn)生方法、裝置、電子終端及介質(zhì)

有效

• 申請(qǐng)?zhí)枺?/td>: 201911045095.8
• 申請(qǐng)日: 2019-10-30
• 公開（公告）號(hào): CN110837786B
• 公開（公告）日: 2022-07-08
• 發(fā)明（設(shè)計(jì)）人: 袁德勝;王作輝;張宏俊;王海濤;姚磊;楊進(jìn)參;游浩泉;劉耀文;吳賀豐
• 申請(qǐng)（專利權(quán)）人: 匯納科技股份有限公司
• 主分類號(hào): G06V20/52
• 分類號(hào): G06V20/52;G06V10/44;G06N3/04
• 代理公司: 上海光華專利事務(wù)所(普通合伙) 31219
• 代理人: 龐紅芳

本發(fā)明提供一種基于空間通道的密度圖產(chǎn)生方法、裝置、電子終端及介質(zhì)，該方法包括：對(duì)輸入的原始圖像進(jìn)行特征提取，以獲得初步特征圖；根據(jù)多個(gè)分割尺度對(duì)初步特征圖進(jìn)行分割，且保留各分割塊中最大的值，獲得對(duì)應(yīng)多個(gè)分割尺度的中間特征圖；將中間特征圖的任意兩點(diǎn)特征之間的位置關(guān)聯(lián)結(jié)果和各通道間的通道關(guān)聯(lián)結(jié)果進(jìn)行融合，獲得位置通道注意力特征圖；對(duì)位置通道注意力特征圖進(jìn)行特征圖插值操作和相接操作，獲取多個(gè)輸出特征圖；將多個(gè)輸出特征圖在通道維度相接，獲得最終特征圖；對(duì)最終特征圖進(jìn)行處理，以獲得密度圖。本發(fā)明在特征的空間維度和通道維度分別抓取特征之間的全局依賴關(guān)系，增強(qiáng)特征的表達(dá)能力，提高密度圖的精度和有效性。

技術(shù)領(lǐng)域

本發(fā)明涉及圖像處理領(lǐng)域，特別是涉及一種基于空間通道的密度圖產(chǎn)生方法、裝置、電子終端及介質(zhì)。

背景技術(shù)

人群計(jì)數(shù)在視頻監(jiān)控、交通監(jiān)測(cè)、公共安全、城市規(guī)劃以及建設(shè)智能商超等方面有著廣泛應(yīng)用，如監(jiān)控某個(gè)人群易聚集區(qū)域的人群數(shù)目，防止由于人群密度過大，導(dǎo)致人群失控發(fā)生踩踏等事件。人群計(jì)數(shù)的結(jié)果直接影響后續(xù)的策略的執(zhí)行，所以需要提出一種準(zhǔn)確且有效的人群密度計(jì)算方法。

發(fā)明內(nèi)容

鑒于以上所述現(xiàn)有技術(shù)的缺點(diǎn)，本發(fā)明的目的在于提供一種基于空間通道的密度圖產(chǎn)生方法、裝置、電子終端及介質(zhì)，用于解決現(xiàn)有技術(shù)中不能準(zhǔn)確且有效的對(duì)人群密度進(jìn)行計(jì)算的問題。

為實(shí)現(xiàn)上述目的及其他相關(guān)目的，本發(fā)明提供一種基于空間通道的密度圖產(chǎn)生方法，包括：對(duì)輸入的原始圖像進(jìn)行特征提取，以獲得初步特征圖；根據(jù)預(yù)設(shè)的多個(gè)分割尺度對(duì)所述初步特征圖進(jìn)行分割，且保留各分割塊中最大的值，以獲得對(duì)應(yīng)所述多個(gè)分割尺度的中間特征圖；將所述中間特征圖的任意兩點(diǎn)特征之間的位置關(guān)聯(lián)結(jié)果和所述中間特征圖的各通道之間的通道關(guān)聯(lián)結(jié)果進(jìn)行融合，以獲得位置通道注意力特征圖；將各所述分割尺度對(duì)應(yīng)的位置通道注意力特征圖進(jìn)行合并；所述合并包括：對(duì)所述位置通道注意力特征圖根據(jù)所述對(duì)應(yīng)的分割尺度進(jìn)行特征圖插值操作，以獲取與所述初步特征圖大小相同的多個(gè)輸出特征圖，且將所述的多個(gè)輸出特征圖在通道維度相接，獲得最終特征圖；根據(jù)具有預(yù)設(shè)卷積尺度的卷積層對(duì)所述最終特征圖進(jìn)行處理，以獲得與所述原始圖像大小相同的密度圖。

在一些實(shí)施例中，根據(jù)預(yù)設(shè)的多個(gè)不同的分割尺度對(duì)所述初步特征圖進(jìn)行分割，且保留各分割塊中最大的值，以獲得對(duì)應(yīng)所述多個(gè)分割尺度的中間特征圖。

在一些實(shí)施例中，將各所述分割尺度對(duì)應(yīng)的位置通道注意力特征圖進(jìn)行合并的步驟包括：分別對(duì)各所述分割尺度對(duì)應(yīng)的位置通道注意力特征圖進(jìn)行插值，以使各所述分割尺度對(duì)應(yīng)的位置通道注意力特征圖的大小相同；且對(duì)進(jìn)行插值后的各所述位置通道注意力特征圖逐次相加；對(duì)逐次相加后獲得的位置通道注意力特征圖再次進(jìn)行插值，以獲得與所述初步特征圖大小相同的最終特征圖。

在一些實(shí)施例中，根據(jù)具有預(yù)設(shè)卷積尺度的卷積層對(duì)所述最終特征圖進(jìn)行處理的步驟包括：根據(jù)所述卷積層對(duì)所述最終特征圖進(jìn)行降維處理，把具有多通道的所述最終特征圖將到一個(gè)通道，且對(duì)一個(gè)通道的所述最終特征圖執(zhí)行圖像插值，以獲得與所述原始圖像大小相同的密度圖。

為實(shí)現(xiàn)上述目的及其他相關(guān)目的，本發(fā)明還提供一種基于空間通道的密度圖產(chǎn)生裝置，包括：特征提取模塊，用以對(duì)輸入的原始圖像進(jìn)行特征提取，以獲得初步特征圖；圖像分割模塊，用以根據(jù)預(yù)設(shè)的多個(gè)分割尺度對(duì)所述初步特征圖進(jìn)行分割，且保留各分割塊中最大的值，以獲得對(duì)應(yīng)所述多個(gè)分割尺度的中間特征圖；注意力模塊，用以將所述中間特征圖的任意兩點(diǎn)特征之間的位置關(guān)聯(lián)結(jié)果和所述中間特征圖的各通道之間的通道關(guān)聯(lián)結(jié)果進(jìn)行融合，以獲得位置通道注意力特征圖；合并模塊，用以將各所述分割尺度對(duì)應(yīng)的位置通道注意力特征圖進(jìn)行合并；所述合并包括：對(duì)所述位置通道注意力特征圖根據(jù)所述對(duì)應(yīng)的分割尺度進(jìn)行特征圖插值操作，以獲取與所述初步特征圖大小相同的多個(gè)輸出特征圖，且用以將所述多個(gè)輸出特征圖在通道維度相接，獲得最終特征圖；估計(jì)模塊，用以根據(jù)具有預(yù)設(shè)卷積尺度的卷積層對(duì)所述最終特征圖進(jìn)行處理，以獲得與所述原始圖像大小相同的密度圖。