search

chrischoy / DeepGlobalRegistration

[CVPR 2020 Oral] A differentiable framework for 3D registration
Other
468 stars 85 forks source link

Output tensor of ME.SparseTensor has different size with input tensor. #13

Open Dangzheng opened 3 years ago

Dangzheng commented 3 years ago

Hi,

Thanks for sharing the code. I'm trying to fine-tune DGR on another dataset. So I modify the dataloader which produces same format data as the data provided by threedmatch_loader. But I found that the output of the ME.SparseTensor has different shape with the input tensor.

To be specific the error lies in this line. https://github.com/chrischoy/DeepGlobalRegistration/blob/46dd264580b4191accedc277f4ae434acdb4d380/core/trainer.py#L633

image

Here is the shape of the tensors, I notice that there are a lot of lines are the same in the tensor iC0. The replicate line issue also happens in the threedmatch data, but the output and the input is same. Do you have any clue about how to solve this issue? Thanks in advance.

Best,

cjluLeiZhu commented 3 years ago

Hi! I am facing a similar issue as well. How did you solve this?

Thanks! @chrischoy @Dangzheng