search

HuguesTHOMAS / KPConv-PyTorch

Kernel Point Convolution implemented in PyTorch
MIT License
789 stars 158 forks source link

the huge time required by segmentation_inputs function #226

Closed cppcute-pm closed 1 year ago

cppcute-pm commented 1 year ago

Hello, Hugues THOMAS. When I try to run the KPConv module, I found that the segmentation_inputs function takes 2.7s-3.0s to generate the input data. I' m wondering whether it's normal to cost so much time or I may make mistakes when I install the related enviroment. And I noticed that your getitem function assum that there are several process loading the data at the same time, and it plays the role of forming a batch of data. I guess these measures you took are trying to make up for the time overhead of the segmentation_inputs function. Am I right?

cppcute-pm commented 1 year ago

My debug code have some issues, in fact the segmentation function doesn't take much time.