Skip to content

GitLab

Unverified Commit 7bc7e551 authored by Shaoshuai Shi's avatar Shaoshuai Shi Committed by GitHub
Browse files

Bugfixed: re-sample gt_boxes when training

parent 0e84f82d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
pcdet/datasets/dataset.py
View file @ 7bc7e551
...@@ -139,7 +139,7 @@ class DatasetTemplate(torch_data.Dataset): ...@@ -139,7 +139,7 @@ class DatasetTemplate(torch_data.Dataset):
data_dict=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__()) new_index = np.random.randint(self.__len__())
return self.__getitem__(new_index) return self.__getitem__(new_index)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment