基于圖神經網絡和注意力機制的點云分類模型

摘要： 為了增強基于深度學習的三維點云分類模型對全局特征的建模能力，提高模型的泛化性能，在PointNet的基礎上，提出了基于圖神經網絡和注意力機制融合的點云分類模型。首先，將提取的特征分別通過增加通道注意力模塊和空間注意力模塊，使模型更加關注全局上下文信息，抑制噪聲信息，減少冗余參數，增強對全局特征的建模能力；其次，通過在多尺度球半徑內進行不同K值最近鄰搜索對編碼的輸入特征進行構圖，...