Computer Science ›› 2024, Vol. 51 ›› Issue (2): 142-150.doi: 10.11896/jsjkx.230200073

• Computer Graphics & Multimedia • Previous Articles     Next Articles

Multi-guided Point Cloud Registration Network Combined with Attention Mechanism

LIU Xuheng, BAI Zhengyao, XU Zhu, DU Jiajin, XIAO Xiao   

  1. School of Information Science and Engineering,Yunnan University,Kunming 650221,China
  • Received:2023-02-20 Revised:2023-06-20 Online:2024-02-15 Published:2024-02-22
  • About author:LIU Xuheng,born in 1998,postgra-duate.His main research interests include point cloud registration and three-dimensional reconstruction.BAI Zhengyao,born in 1967,Ph.D,professor,master supervisor.His main research interests include signal proces-sing,image processing,pattern recognition and machine learning,etc.
  • Supported by:
    Yunnan Provincial Major Science and Technology Special Plan Projects(202002AD080001) and 14th Postgra-duate Research Innovation Project of Yunnan University(KC-22222543).

Abstract: This paper proposes a point cloud alignment network,AMGNet,which uses the probability matrix of matching points between point clouds and the spatial information feature matrix of point clouds to search for correspondence and determine the weights of corresponding points with each other.First,the point cloud feature extraction network is used to get the high-dimensional features of the two unaligned point clouds and then the Transformer is used to fuse the independent features with the contextual information.Also,the weight assignment uses the strategy of double matrix co-determination.Finally,the singular value decomposition is used to obtain the required rigid transformation matrix.Several experiments are conducted on synthetic datasets,such as ModelNet40,7Scenes and real scenes.The results show that the mean square error of rotation matrix and translation vector in ModelNet40 target unknown experiments is reduced to 0.025 and 0.004 6,respectively.AMGNet alignment has high accuracy,high interference resistance,and good generalization ability.

Key words: Point cloud registration, Attention mechanism, Multiple matrix guidance, Weighted SVD

CLC Number: 

  • TP391.41
[1]LI Z M,ZHANG Y P,LIU Y J,et al.Deformable Graph Convolutional Networks Based Point Cloud Representation Learning[J].Computer Science,2022,49(8):273-278.
[2]HUANG X,MEI G,ZHANG J,et al.A comprehensive survey on point cloud registration[J].arXiv:2103.02690,2021.
[3]QIN H X,LIU Z T,TAN B Y.Review on deep learning rigid point cloud registration[J].Journal of Image and Graphics,2022,27(2):329-348.
[4]LI J W,ZHAN J W.Review on 3D point cloud registrationmethod[J].Journal of Image and Graphics,2022,27(2):349-367.
[5]LI J,ZHANG C,XU Z,et al.Iterative distance-aware similarity matrix convolution with mutual-supervised point elimination for efficient point cloud registration[C]//European Conference on Computer Vision.Berlin:Springer,2020:378-394.
[6]WANG H,LIU X,KANG W,et al.Multi-features guidance net-work for partial-to-partial point cloud registration[J].Neural Computing and Applications,2022,34(2):1623-1634.
[7]HEZRONI I,DRORY A,GIRYES R,et al.DeepBBS:Deep Best Buddies for Point Cloud Registration[C]//2021 International Conference on 3DVision(3DV).Piscataway,NJ:IEEE.2021:342-351.
[8]WANG Y,SOLOMON J M.Deep closest point:Learning rep resentations for point cloud registration[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision.Piscataway,NJ:IEEE 2019:3523-3532.
[9]YEW Z J,LEE G H.RPM-Net:Robust Point Matching usingLearned Features[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).Piscataway,NJ:IEEE,2020:11821-11830.
[10]BESL P J,MCKAY N D.Method for registration of 3-D shapes[C]//Sensor Fusion IV:Control Paradigms and Data Structures.Spie,1992:586-606.
[11]YANG J,LI H,CAMPBALL D,et al.Go-ICP:A globally optimal solution to 3D ICP point-set registration[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2015,38(11):2241-2254.
[12]HEXSEL B,VHAVLE H,CHEN Y.DICP:Doppler IterativeClosest Point Algorithm[J].arXiv:2201.11944,2022.
[13]MAGNUSSON M,LILIENTHAL A,DUCKETT T.Scan registration for autonomous mining vehicles using 3D-NDT[J].Journal of Field Robotics,2007,24(10):803-827.
[14]AIGER D,MITRA N J,COHEN-OR D.4-points congruent sets for robust pairwise surface registration[J].ACM Transactions on Graphics,2008,27(3):1-10.
[15]FISCHLER M A,BOLLES R C.Random sample consensus:aparadigm for model fitting with applications to image analysis and automated cartography[J].Communications of the ACM,1981,24(6):381-395.
[16]MELLADO N,AIGER D,MITRA N J.Super 4PCS Fast Global Pointcloud Registration via Smart Indexing[J].Computer Graphics Forum,2014,33(5):205-215.
[17]KAMOUSI P,LAZARD S,MAHESHWARI A,et al.Analysis of farthest point sampling for approximating geodesics in a graph[J].Computational Geometry,2016,57:1-7.
[18] RUSINKIEWICZ S,LEVOV M.Efficient variants of the ICP algorithm[C]//Proceedings third International Conference on 3-D Digital Imaging and Modeling.Piscataway,NJ:IEEE,2001:145-152.
[19]PRAKHYA S M,LIU B,LIN W.Detecting keypoint sets on 3D point clouds via Histogram of Normal Orientations[J].Pattern Recognition Letters,2016,83:42-48.
[20]CHUA C S.Point Signatures:A New Representation for 3D Object Recognition[J].International Journal of Computer Vision,1997,25:63-85.
[21]FROME A,HUBER D,KOLLURI R,et al.Recognizing objects in range data using regional point dscriptors[C]//European Conference on Computer Vision.Berlin:Springer:2004:224-237.
[22]RUSU R B,BLODOW N,MARTON Z C,et al.Aligning point cloud views using persistent feature histograms[C]//2008 IEEE/RSJ International Conference on Intelligent Robots and Systems.Piscataway,NJ:IEEE.2008:3384-3391.
[23]RUSU R B,BLODOW N,BEETZ M.Fast point feature histo-gra.ms(FPFH) for 3D registration[C]//2009 IEEE International Conference on Robotics and Automation.Piscataway,NJ:IEEE,2009:3212-3217.
[24] ZHANG W L,CHENG L,REN M F,et al.Point Cloud Registration Based on AGConv Local Feature Descriptors[J].Computer Engineering,2023,49(11):231-237.
[25] LI X M,WANG C Y,LIU Xl,et al.Point cloud registration method based on supervoxel bidirectional nearest neighbor distance ratio[J].Journal of Jilin University(Engineering and Technology Edition),2022,52(8):1918-1925.
[26]AOKI Y,GOFORTH H,SRIVATSAN R A,et al.Pointnetlk:Robust & efficient point cloud registration using pointnet[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition.Piscataway,NJ:IEEE,2019:7163-7172.
[27]LU W,WAN G,ZHOU Y,et al.Deepvcp:An end-to-end deep neural network for point cloud registration[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision.Piscataway,NJ:IEEE,2019:12-21.
[28]MONTI F,BOSCAINI D,MASCI J,et al.Geometricdeep lear-ning on graphs and manifolds using mixture model cnns[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition(CVPR).Piscataway,NJ:IEEE,2017:5425-5434.
[29]CHARLES R Q,SU H,KAICHUN M,et al.PointNet:Deep Learning on Point Sets for 3D Classification and Segmentation[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition(CVPR).Piscataway,NJ:IEEE,2017:77-85.
[30]WANG Y,SUN Y,LIU Z,et al.Dynamic graph cnn for learning on point clouds[J].ACM Transactions on Graphics(TOG),2019,38(5):1-12.
[31]QI C R,YI L,SU H,et al.Pointnet++:Deep hierarchical feature learning on point sets in a metric space[J].arXiv:1706.02413,2017.
[32]VASWANI A,SHAZEER N,PARMAR N,et al.Attention isall you need[J].arXiv:1706.03762,2017.
[33]LI J,ZHANG C,XU Z,et al.Iterative distance-aware similarity matrix convolution with mutual-supervised point elimination for efficient point cloud registration[C]//Computer Vision-ECCV 2020:16th European Conference,Part XXIV 16.Springer International Publishing,2020:378-394.
[34]WU Z R,SONG S,KHOSLA A,et al.3D ShapeNets:A deep representation for volumetric shapes[C]//2015 IEEE Confe-rence on Computer Vision and Pattern Recognition(CVPR).Piscataway,NJ:IEEE,2015:1912-1920.
[35] SHOTTON J,GLOCKER B,ZACH C,et al.Scene coordinate regression forests for camera relocalization in RGB-D images[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Piscataway,NJ:IEEE,2013:2930-2937.
[36] ZHOU Q Y,PARK J,KOLTUN V.Open3D:A modern library for 3D data processing[J].arXiv:1801.09847,2018.
[37]SHEN Y,HUI L,JIANG H,et al.Reliable Inlier Evaluation for Unsupervised Point Cloud Registration[C]//Proceedings of the AAAI Conference on Artificial Intelligence.California:Association for the Advancement of Artificial Intelligence(AAAI),2022:2198-2206.
[1] WANG Ruiping, WU Shihong, ZHANG Meihang, WANG Xiaoping. Review of Vision-based Neural Network 3D Dynamic Gesture Recognition Methods [J]. Computer Science, 2024, 51(4): 193-208.
[2] ZHANG Mingdao, ZHOU Xin, WU Xiaohong, QING Linbo, HE Xiaohai. Unified Fake News Detection Based on Semantic Expansion and HDGCN [J]. Computer Science, 2024, 51(4): 299-306.
[3] WANG Zihong, SHAO Yingxia, HE Jiyuan, LIU Jinbao. Sequential Recommendation Based on Multi-space Attribute Information Fusion [J]. Computer Science, 2024, 51(3): 102-108.
[4] HAO Ran, WANG Hongjun, LI Tianrui. Deep Neural Network Model for Transmission Line Defect Detection Based on Dual-branch Sequential Mixed Attention [J]. Computer Science, 2024, 51(3): 135-140.
[5] LI Yu, YANG Xiangli, ZHANG Le, LIANG Yalin, GAO Xian, YANG Jianxi. Combined Road Segmentation and Contour Extraction for Remote Sensing Images Based on Cascaded U-Net [J]. Computer Science, 2024, 51(3): 174-182.
[6] LIAO Meng, JIA Zhen, LI Tianrui. Chinese Named Entity Recognition Based on Label Information Fusion and Multi-task Learning [J]. Computer Science, 2024, 51(3): 198-204.
[7] SUN Shounan, WANG Jingbin, WU Renfei, YOU Changkai, KE Xifan, HUANG Hao. TMGAT:Graph Attention Network with Type Matching Constraint [J]. Computer Science, 2024, 51(3): 235-243.
[8] ZHANG Guodong, CHEN Zhihua, SHENG Bin. Infrared Small Target Detection Based on Dilated Convolutional Conditional GenerativeAdversarial Networks [J]. Computer Science, 2024, 51(2): 151-160.
[9] ZHANG Feng, HUANG Shixin, HUA Qiang, DONG Chunru. Novel Image Classification Model Based on Depth-wise Convolution Neural Network andVisual Transformer [J]. Computer Science, 2024, 51(2): 196-204.
[10] LIAO Xingbin, QIAN Yangge, WANG Qianlei, QIN Xiaolin. Hierarchical Document Classification Method Based on Improved Self-attention Mechanism and Representation Learning [J]. Computer Science, 2024, 51(2): 238-244.
[11] WANG Weijia, XIONG Wenzhuo, ZHU Shengjie, SONG Ce, SUN He, SONG Yulong. Method of Infrared Small Target Detection Based on Multi-depth Feature Connection [J]. Computer Science, 2024, 51(1): 175-183.
[12] SHI Dianxi, LIU Yangyang, SONG Linna, TAN Jiefu, ZHOU Chenlei, ZHANG Yi. FeaEM:Feature Enhancement-based Method for Weakly Supervised Salient Object Detection via Multiple Pseudo Labels [J]. Computer Science, 2024, 51(1): 233-242.
[13] BAI Zhengyao, XU Zhu, ZHANG Yihan. Deep Artificial Correspondence Generation for 3D Point Cloud Registration [J]. Computer Science, 2023, 50(9): 210-219.
[14] YI Liu, GENG Xinyu, BAI Jing. Hierarchical Multi-label Text Classification Algorithm Based on Parallel Convolutional Network Information Fusion [J]. Computer Science, 2023, 50(9): 278-286.
[15] LUO Yuanyuan, YANG Chunming, LI Bo, ZHANG Hui, ZHAO Xujian. Chinese Medical Named Entity Recognition Method Incorporating Machine ReadingComprehension [J]. Computer Science, 2023, 50(9): 287-294.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!