search

liuzhengzhe / One-Thing-One-Click

https://arxiv.org/abs/2104.02246 One Thing One Click (CVPR 2021) https://arxiv.org/abs/2303.14727 One Thing One Click++ (Arxiv)
53 stars 8 forks source link

Tensor Sizes don't match #9

Closed derkaczda closed 2 years ago

derkaczda commented 2 years ago

Thanks for the work! I am trying to test the code like written in the README.md 

cd 3D-U-Net
python test.py --config config/pointgroup_run1_scannet.yaml --pretrain pointgroup_run1_scannet-000001250.pth

But I get following error

RuntimeError: The size of tensor a (81369) must match the size of tensor b (80583) at non-singleton dimension 0

Why are the scores accumulated in the first place? Would be nice if someone could help