Merge pull request open-mmlab#513 from jihanyang/master

fixbug: multi_classes_nms and waymo dataset with old version splits
ChienLiu · Apr 20, 2021 · e4d2b75 · e4d2b75
2 parents 0642cf0 + c25f699
commit e4d2b75
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pcdet/datasets/waymo/waymo_dataset.py b/pcdet/datasets/waymo/waymo_dataset.py
@@ -70,7 +70,7 @@ def include_waymo_data(self, mode):
     @staticmethod
     def check_sequence_name_with_all_version(sequence_file):
         if '_with_camera_labels' not in str(sequence_file) and not sequence_file.exists():
-            sequence_file = Path(str(sequence_file[:-9]) + '_with_camera_labels.tfrecord')
+            sequence_file = Path(str(sequence_file)[:-9] + '_with_camera_labels.tfrecord')
         if '_with_camera_labels' in str(sequence_file) and not sequence_file.exists():
             sequence_file = Path(str(sequence_file).replace('_with_camera_labels', ''))
 

diff --git a/pcdet/models/model_utils/model_nms_utils.py b/pcdet/models/model_utils/model_nms_utils.py
@@ -44,6 +44,7 @@ def multi_classes_nms(cls_scores, box_preds, nms_config, score_thresh=None):
             cur_box_preds = box_preds[scores_mask]
         else:
             box_scores = cls_scores[:, k]
+            cur_box_preds = box_preds
 
         selected = []
         if box_scores.shape[0] > 0: