Bugfixed: re-sample gt_boxes when training

7bc7e551 · Shaoshuai Shi · GitHub · 0e84f82d · 7bc7e551
Unverified Commit 7bc7e551 authored Nov 08, 2020 by Shaoshuai Shi Committed by GitHub Nov 08, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pcdet/datasets/dataset.py pcdet/datasets/dataset.py +1 -1

No files found.
--- a/pcdet/datasets/dataset.py
+++ b/pcdet/datasets/dataset.py
@@ -139,7 +139,7 @@ class DatasetTemplate(torch_data.Dataset):
            data_dict=data_dict
        )

-        if len(data_dict['gt_boxes']) == 0:
+        if self.training and len(data_dict['gt_boxes']) == 0:
            new_index = np.random.randint(self.__len__())
            return self.__getitem__(new_index)