RegTR icon indicating copy to clipboard operation
RegTR copied to clipboard

I want test your own data, how many points can you enter at most each time?

Open wang89280 opened this issue 1 year ago • 2 comments

I am now using the pre-trained model to predict transformation.

My data are 8000 and 7000 respectively, but when using the model for inference, but it shows that the memory is out of bounds. How should I solve this problem in the code or data.?

My graphics card is RTX3090, changing the graphics card is too expensive for me.

wang89280 avatar Dec 06 '23 08:12 wang89280

24GB is sufficient for inference.

The memory heavy portion in the network is the attention layers which takes place only in the coarsest downsampled level. We designed the network to downsample to a few hundred points (~500) in this coarse level, and I suspect that your coarse point cloud has much more points than this. This is due to the point clouds having different scales.

If that is the case, you need to configure the KPConv layers to suit your dataset. Particularly, you may have to adjust the *_radius fields in the configuration file. Nevertheless, since your point clouds are very different from the training dataset, you might not get good results without retraining the network.

yewzijian avatar Dec 11 '23 04:12 yewzijian

Cloud you tell me how to modify this code in order to correct result? The data is in the attachment. des.txt src.txt

wang89280 avatar Dec 15 '23 02:12 wang89280