luost26 / DMRDenoise

:snowman: Differentiable Manifold Reconstruction for Point Cloud Denoising (ACM MM 2020)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

error when test on custom data

JANEKKO opened this issue · comments

Hi @luost26 ,Thank you for sharing this code.. I'm trying to test my own data and it is reported of this error:
44%|████████████████████████████████████████████████████▎ | 11/25 [00:01<00:02, 6.00it/s] 3%|███▏ | 3/113 [00:36<22:14, 12.13s/it] Traceback (most recent call last): File "denoise.py", line 202, in <module> auto_denoise(args) File "denoise.py", line 154, in auto_denoise expand_knn=args.expand_knn File "denoise.py", line 116, in run_denoise_large_pointcloud den, dow = run_denoise(patch - centers[i], patch_size, ckpt, device, random_state, expand_knn) File "denoise.py", line 53, in run_denoise pred = model(patch) File "/home/xxx/anaconda3/envs/DMRDenoise/lib/python3.6/site-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/home/xxx/DMRDenoise-master/models/denoise.py", line 56, in forward return self.model(pos) File "/home/xxx/anaconda3/envs/DMRDenoise/lib/python3.6/site-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/home/xxx/DMRDenoise-master/models/net.py", line 54, in forward feats.append(feat_unit(pos)) File "/home/xxx/anaconda3/envs/DMRDenoise/lib/python3.6/site-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/home/xxx/DMRDenoise-master/models/blocks.py", line 48, in forward return self.dynamic_graph_forward(x) File "/home/xxx/DMRDenoise-master/models/blocks.py", line 36, in dynamic_graph_forward x = self.convs[i](x, x) File "/home/xxx/anaconda3/envs/DMRDenoise/lib/python3.6/site-packages/torch/nn/modules/module.py", line 550, in __call__ result = self.forward(*input, **kwargs) File "/home/xxx/DMRDenoise-master/models/conv.py", line 54, in forward ], dim=-1) # (B, N, K, d+c) RuntimeError: Sizes of tensors must match except in dimension 2. Got 24 and 19

I wonder ,Whether there's any requirement for the test data? I used .txt file , with x, y, z coordinates per line. Thank you , Looking forward to your reply.