From ca0faaf047aaec1adc66fb18c199f6dc4316cfba Mon Sep 17 00:00:00 2001 From: Eric Wiener Date: Tue, 19 Jan 2021 02:02:13 -0500 Subject: [PATCH] Fixed issue unpacking size in furthest_point_sample (#248) 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)