Test_shape failed with torchsparse 1.2

[bingting1022]

Good work! I manged to get the shape using Test_shape.py but when using test_shape.py, I faced this error:
processing (0000)-th image... LeReS/test_images/5.jpg
Traceback (most recent call last):
File "LeReS/tools/test_shape.py", line 123, in
shift, focal_length, depth_scaleinv = reconstruct3D_from_depth(rgb, pred_depth_ori,
File "LeReS/tools/test_shape.py", line 76, in reconstruct3D_from_depth
shift_1 = refine_shift(pred_depth_norm, shift_model, predicted_focal_1, cam_u0, cam_v0)
File "/home/adelai_ws/AdelaiDepth/LeReS/lib/test_utils.py", line 124, in refine_shift
shift = refine_shift_one_step(depth_wshift_tmp, model, focal, u0, v0)
File "/home/adelai_ws/AdelaiDepth/LeReS/lib/test_utils.py", line 108, in refine_shift_one_step
outputs = model(inputs)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/trainer/adelai_ws/AdelaiDepth/LeReS/lib/spvcnn_classsification.py", line 148, in forward
x2 = self.stage2(x1)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/container.py", line 141, in forward
input = module(input)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/trainer/adelai_ws/AdelaiDepth/LeReS/lib/spvcnn_classsification.py", line 23, in forward
out = self.net(x)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/container.py", line 141, in forward
input = module(input)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/torchsparse/nn/modules/conv.py", line 72, in forward
return conv3d(inputs,
File "/usr/local/lib/python3.8/dist-packages/torchsparse/nn/functional/conv.py", line 118, in conv3d
idx_query = list(convert_neighbor_map_gpu(idx_query))
File "/usr/local/lib/python3.8/dist-packages/torchsparse/nn/functional/convert_neighbor_map.py", line 9, in forward
idx_batch, idx_point = torch.where(neighbor_map != -1)
ValueError: not enough values to unpack (expected 2, got 1)

[jgallegov]

Same error here...

File "/home/azken/workspace/3D_ROOM_RECONSTRUCTION/AdelaiDepth/venv/lib/python3.7/site-packages/torchsparse/nn/functional/convert_neighbor_map.py", line 21, in forward
neighbor_map.int(), idx_batch.int(), idx_point.int())

[yukiumi13]

#9 (comment) works.

[YvanYin]

@YangLI-aftermath thx.

[YvanYin]

Please check the version of GCC. We used GCC 7.0.

[kamzero]

It is an error with torchsparse 1.2.0
this comment works