一种基于图注意力网络的三维点云分类方法

基本信息

申请号 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 分类 计算;推算;计数;
发明人 吕孟东;俞宏群;吴月超;沈惠良;朱德康;彭礼平;侯进皎;程翔;许鹤鹏;王熙震;傅嘉辉;吕小锋;章伽宁;张学超 申请(专利权)人 国网新源集团有限公司富春江水力发电厂
代理机构 杭州新源专利事务所(普通合伙) 代理人 董晨楠
地址 311504浙江省杭州市桐庐县富春江镇
法律状态 -

摘要

摘要 本发明公开了一种基于图注意力网络的三维点云分类方法,利用构建的图神经网络,完成对原始点云输入数据的点云分类;所述图神经网络包括特征提取层、两组交替分布的图注意力层和池化层、激活函数层、读出层、全连接层和输出层。采用图注意力机制,同时考虑点和边的属性,结合使用图的池化操作,将图逐步变小,构建图神经网络来对三维点云分类,从而可以有效的提高三维点云分类的准确率。采用本发明的方法对ModelNet40数据集(包含40个类别的CAD模型)进行分类处理,经测试,在该数据集上分类正确率为91.5%。