NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
https://nvidia.github.io/MinkowskiEngine
Other
2.48k stars 367 forks source link

Question about the completion task. #360

Closed Lillian9707 closed 3 years ago

Lillian9707 commented 3 years ago

Hi, thanks for the incredible framework for point cloud processing. However, when I read your code on the completion task in examples/completion.py, I noticed that the input of the network is still the complete point cloud:

In line 541, sin = ME.SparseTensor( features=in_feat, coordinates=data_dict["coords"], device=device, ) where data_dict['coords'] refers to point cloud which is not randomly cropped.

I wonder if I have misunderstood the code.

chrischoy commented 3 years ago

Please refer to the transformation class in the same file. It is the cropped data.