PCRNet: Point Cloud Registration Network using PointNet Encoding

最新推荐文章于 2024-02-20 18:08:57 发布

dloading7

最新推荐文章于 2024-02-20 18:08:57 发布

阅读量1.2k

点赞数 4

分类专栏： Registration_ModelNet40 文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/dloading7/article/details/119919307

版权

2 篇文章 0 订阅

订阅专栏

在这里插入图片描述
PCRNet算是在ModelNet40这种object-centric数据集上用deep-learning做配准最好入手的一篇文章了，整体架构涉及简洁明了，一句话描述就是简单粗暴，话不多说，直接上网络架构图：

首先对输入source ( $\times 3$ )与template ( $\times 3$ )进行PointNet操作，即MLP + max-pooling后，分别得到描述各自点云的维度为1024的global feature vector： $\Phi ({P_S}),\Phi ({P_T})$ .
此时可以粗略的认为 $\Phi ({P_S}),\Phi ({P_T})$ 分别encode了 ${P_S}$ 与 ${P_T}$ 各自的位置信息，接着进行concatenate拼接操作将两个1024维向量拼成2048维。
接着直接将配准问题建模为一个简单的回归问题，将concatenate输出了2048维向量应用FC层投影成最终的7维向量输出：前三维代表translation vector，后四维代表rotation matrix转化成的四元素表示。

简单的模型可想而知带来的就是效果不可能会很好，因此一次不行就来多次，反正模型简单做一次inference时间会很快，因此最后使用时是以迭代形式 (Iterative PCRNet)呈现：
在这里插入图片描述

这里使用的loss是点云配准中不经常出现的Earth Mover Distance (EMDLoss)：
在这里插入图片描述
我觉得这种做一般了解就好，毕竟不是Registration中的主流loss.

代码使用的是发哥的代码，整体代码是比较易于阅读的，这里直接贴出我复现的结果：

Method	isotropic R	isotropic t	anisotropic(mse, mae)	anisotropic(mse, mae)	training_mode	epochs	trainset/testset	testing with noise/partial
Iter-PCRNet	6.77	0.09	10.30(3.23)	0.13(0.04)	clean	600	9840/2468	N/N
Iter-PCRNet	7.01	0.09	9.94(3.41)	0.13(0.04)	clean	600	9840/1266	N/N
Iter-PCRNet	27.55	0.39	28.17(15.47)	0.30(0.19)	clean	600	9840/2468	N/Y

分别测试了default、unseen、以及partial三种setting下的表现。

感觉PCRNet里面真正值得研究的东西没多少，但是作为一种在ModelNet40上深度学习配准的入门文章与代码，还是挺不错的。

Pretrained Model can be found at here:link

关注