From b60ab80027907f0a41878c30da62a76939ae01de Mon Sep 17 00:00:00 2001 From: Eric Wiener Date: Fri, 25 Dec 2020 15:33:44 -0500 Subject: [PATCH] Fixed issue unpacking size in furthest_point_sample The tuple unpacked into three arguments when four were given when running VoteNet on sunrgbd --- mmdet3d/ops/furthest_point_sample/furthest_point_sample.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdet3d/ops/furthest_point_sample/furthest_point_sample.py b/mmdet3d/ops/furthest_point_sample/furthest_point_sample.py index d03b57c7ac..af53317810 100644 --- a/mmdet3d/ops/furthest_point_sample/furthest_point_sample.py +++ b/mmdet3d/ops/furthest_point_sample/furthest_point_sample.py @@ -25,7 +25,7 @@ def forward(ctx, points_xyz: torch.Tensor, """ assert points_xyz.is_contiguous() - B, N, _ = points_xyz.size() + B, N = points_xyz.size()[:2] output = torch.cuda.IntTensor(B, num_points) temp = torch.cuda.FloatTensor(B, N).fill_(1e10)