search

yanx27 / PointASNL

PointASNL: Robust Point Clouds Processing using Nonlocal Neural Networks with Adaptive Sampling （CVPR 2020）
MIT License
260 stars 34 forks source link

Performance issues in SemanticKITTI/semantic_kitti_dataset_grid.py(P2) #44

Closed DLPerf closed 3 years ago

DLPerf commented 3 years ago

Hello,I found a performance issue in the definition of init_input_pipeline , SemanticKITTI/semantic_kitti_dataset_grid.py, self.train_data = self.train_data.map was called without num_parallel_calls. I think it will increase the efficiency of your program if you add this.

The same issues also exist in self.test_data = self.test_data.map , self.val_data = self.val_data.map

Here is the documemtation of tensorflow to support this thing.

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.

yanx27 commented 3 years ago

@DLPerf Thanks for your attention. I'm exactly too busy now, and you are more than welcome to create a PR :)