From c25f699c99ea7d5cab3326a1bcff7b0578899efb Mon Sep 17 00:00:00 2001 From: jihanyang Date: Tue, 20 Apr 2021 17:30:18 +0800 Subject: [PATCH] fixbug: multi_classes_nms and waymo dataset with old version splits --- pcdet/datasets/waymo/waymo_dataset.py | 2 +- pcdet/models/model_utils/model_nms_utils.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/pcdet/datasets/waymo/waymo_dataset.py b/pcdet/datasets/waymo/waymo_dataset.py index a18301a34..ffffa8bc4 100644 --- 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 index eb11d2d34..0aeada497 100644 --- 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: