一种基于图注意力网络的三维点云分类方法
基本信息
申请号 | 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%。 |
