本申請(qǐng)實(shí)施例公開(kāi)了一種圖池化方法、裝置、設(shè)備以及存儲(chǔ)介質(zhì)，該方法包括：獲取待池化圖特征表示，并將待池化圖特征表示分割為多個(gè)子圖特征表示，一個(gè)子圖特征表示中包括待池化圖特征表示中的一個(gè)或者多個(gè)維度的圖特征表示；基于圖神經(jīng)網(wǎng)絡(luò)模型確定各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征，一個(gè)子圖特征表示對(duì)應(yīng)一個(gè)圖注意力輸出特征；基于各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征聚合得到待篩選圖特征表示；根據(jù)待篩選圖特征表示中包括的多個(gè)待篩選對(duì)象的特征表示以及各待篩選對(duì)象的權(quán)重，篩選出目標(biāo)圖特征，并基于目標(biāo)圖特征確定池化后的目標(biāo)圖特征表示。采用本發(fā)明實(shí)施例，可按維度分割圖片特征后對(duì)圖片進(jìn)行池化，增加圖池化效率。

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

本申請(qǐng)涉及圖片處理技術(shù)領(lǐng)域，尤其涉及一種圖池化方法、裝置、設(shè)備以及存儲(chǔ)介質(zhì)。

背景技術(shù)

池化(pooling)是一種降低原始輸入的特征表示大小，通過(guò)更少的特征值表示現(xiàn)有輸入的方法。圖池化的目的是縮小原始輸入圖的表征尺寸，通常可以通過(guò)降低特征維度，或者刪去某些節(jié)點(diǎn)和邊來(lái)實(shí)現(xiàn)。現(xiàn)有技術(shù)中，圖池化都是從拓?fù)鋵用鏈p少整個(gè)圖的尺寸來(lái)獲得更緊湊的表示，使用圖神經(jīng)網(wǎng)絡(luò)模型來(lái)獲得節(jié)點(diǎn)過(guò)濾的權(quán)重相比于只使用節(jié)點(diǎn)特征去映射獲得權(quán)重，考慮了節(jié)點(diǎn)之間的相互影響，要求神經(jīng)網(wǎng)絡(luò)模型的規(guī)模較大，計(jì)算數(shù)據(jù)量大，圖池化效率低。

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

本申請(qǐng)實(shí)施例提供一種圖池化方法、裝置、設(shè)備以及存儲(chǔ)介質(zhì)，可按維度分割圖片特征后對(duì)圖片進(jìn)行池化，增加圖池化效率。

第一方面，本申請(qǐng)實(shí)施例提供一種圖池化方法，該方法包括：

獲取待池化圖特征表示，并將上述待池化圖特征表示分割為多個(gè)子圖特征表示，一個(gè)子圖特征表示中包括上述待池化圖特征表示中的一個(gè)或者多個(gè)維度的圖特征表示；

基于圖神經(jīng)網(wǎng)絡(luò)模型確定各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征，一個(gè)子圖特征表示對(duì)應(yīng)一個(gè)圖注意力輸出特征；

基于上述各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征聚合得到待篩選圖特征表示，上述待篩選圖特征表示中包括上述待池化圖特征表示中的各個(gè)維度對(duì)應(yīng)的特征表示；

根據(jù)上述待篩選圖特征表示中包括的多個(gè)待篩選對(duì)象的特征表示以及各待篩選對(duì)象的權(quán)重，從上述待篩選圖特征中篩選出目標(biāo)圖特征，并基于上述目標(biāo)圖特征確定池化后的目標(biāo)圖特征表示。

結(jié)合第一方面，在一種可能的實(shí)施方式中，上述基于上述各子圖特征表示對(duì)應(yīng)的圖注意力輸出特征聚合得到待篩選圖特征表示之后，上述方法還包括：

利用權(quán)重函數(shù)對(duì)上述待篩選圖特征表示中包括的各待篩選對(duì)象的特征表示進(jìn)行計(jì)算，得到上述各待篩選對(duì)象的權(quán)重。

結(jié)合第一方面，在一種可能的實(shí)施方式中，上述根據(jù)上述待篩選圖特征表示中包括的多個(gè)待篩選對(duì)象的特征表示以及各待篩選對(duì)象的權(quán)重，從上述待篩選圖特征中篩選出目標(biāo)圖特征包括：

將上述待篩選圖特征表示中包括的多個(gè)待篩選對(duì)象按照各待篩選對(duì)象的權(quán)重進(jìn)行排序，將權(quán)重大于權(quán)重閾值的待篩選對(duì)象確定為目標(biāo)對(duì)象；

根據(jù)上述目標(biāo)對(duì)象確定目標(biāo)圖特征；

其中，上述待篩選對(duì)象包括節(jié)點(diǎn)和/或邊。

結(jié)合第一方面，在一種可能的實(shí)施方式中，上述根據(jù)上述目標(biāo)對(duì)象，確定出上述目標(biāo)圖特征，包括：

根據(jù)上述目標(biāo)對(duì)象確定目標(biāo)對(duì)象索引，基于上述目標(biāo)對(duì)象索引生成目標(biāo)對(duì)象掩膜；

根據(jù)上述目標(biāo)對(duì)象掩膜以及上述待篩選圖特征表示確定目標(biāo)圖特征。

結(jié)合第一方面，在一種可能的實(shí)施方式中，上述基于上述目標(biāo)圖特征確定池化后的目標(biāo)圖特征表示之后，上述方法還包括：

基于上述池化后的目標(biāo)圖特征表示，確定池化后的目標(biāo)圖片并輸出上述目標(biāo)圖片。