Bugfixed: re-sample gt_boxes when training

TRAILab · Nov 7, 2020 · 7bc7e55 · 7bc7e55
1 parent 0e84f82
commit 7bc7e55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pcdet/datasets/dataset.py b/pcdet/datasets/dataset.py
@@ -139,7 +139,7 @@ def prepare_data(self, data_dict):
             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)