一種基于圖注意力網(wǎng)絡(luò)的三維點云分類方法

基本信息

申請?zhí)?/td> CN202110900943.X 申請日 -
公開（公告）號 CN113592013A 公開（公告）日 2021-11-02
申請公布號 CN113592013A 申請公布日 2021-11-02
分類號 G06K9/62（2006.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 呂孟東;俞宏群;吳月超;沈惠良;朱德康;彭禮平;侯進皎;程翔;許鶴鵬;王熙震;傅嘉輝;呂小鋒;章伽寧;張學超 申請（專利權(quán)）人 國網(wǎng)新源集團有限公司富春江水力發(fā)電廠
代理機構(gòu) 杭州新源專利事務(wù)所（普通合伙） 代理人 董晨楠
地址 311504浙江省杭州市桐廬縣富春江鎮(zhèn)
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于圖注意力網(wǎng)絡(luò)的三維點云分類方法，利用構(gòu)建的圖神經(jīng)網(wǎng)絡(luò)，完成對原始點云輸入數(shù)據(jù)的點云分類；所述圖神經(jīng)網(wǎng)絡(luò)包括特征提取層、兩組交替分布的圖注意力層和池化層、激活函數(shù)層、讀出層、全連接層和輸出層。采用圖注意力機制，同時考慮點和邊的屬性，結(jié)合使用圖的池化操作，將圖逐步變小，構(gòu)建圖神經(jīng)網(wǎng)絡(luò)來對三維點云分類，從而可以有效的提高三維點云分類的準確率。采用本發(fā)明的方法對ModelNet40數(shù)據(jù)集（包含40個類別的CAD模型）進行分類處理，經(jīng)測試，在該數(shù)據(jù)集上分類正確率為91.5％。