From d8cc12b0cb5e3c8b4661f77f2a45fbd8f5c92db8 Mon Sep 17 00:00:00 2001 From: RangiLyu Date: Mon, 28 Nov 2022 15:58:29 +0800 Subject: [PATCH 1/2] [Fix] Fix browse dataset. --- mmdet/visualization/local_visualizer.py | 2 ++ tools/analysis_tools/browse_dataset.py | 1 + 2 files changed, 3 insertions(+) diff --git a/mmdet/visualization/local_visualizer.py b/mmdet/visualization/local_visualizer.py index 01165a79108..6ee495afe1f 100644 --- a/mmdet/visualization/local_visualizer.py +++ b/mmdet/visualization/local_visualizer.py @@ -335,6 +335,7 @@ def add_datasample( and masks. Defaults to 0.3. step (int): Global step value to record. Defaults to 0. """ + image = image.clip(0, 255).astype(np.uint8) classes = self.dataset_meta.get('CLASSES', None) palette = self.dataset_meta.get('PALETTE', None) @@ -377,6 +378,7 @@ def add_datasample( classes) if gt_img_data is not None and pred_img_data is not None: + print(gt_img_data.dtype, pred_img_data.dtype) drawn_img = np.concatenate((gt_img_data, pred_img_data), axis=1) elif gt_img_data is not None: drawn_img = gt_img_data diff --git a/tools/analysis_tools/browse_dataset.py b/tools/analysis_tools/browse_dataset.py index e4035c0a2bf..b9d90ab1fc1 100644 --- a/tools/analysis_tools/browse_dataset.py +++ b/tools/analysis_tools/browse_dataset.py @@ -78,6 +78,7 @@ def main(): osp.basename(img_path), img, data_sample, + draw_pred=False, show=not args.not_show, wait_time=args.show_interval, out_file=out_file) From 8f533e189938e86b06baa79179341b586a3435f0 Mon Sep 17 00:00:00 2001 From: RangiLyu Date: Mon, 28 Nov 2022 16:25:11 +0800 Subject: [PATCH 2/2] clean code --- mmdet/visualization/local_visualizer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mmdet/visualization/local_visualizer.py b/mmdet/visualization/local_visualizer.py index 6ee495afe1f..62f99f8653e 100644 --- a/mmdet/visualization/local_visualizer.py +++ b/mmdet/visualization/local_visualizer.py @@ -378,7 +378,6 @@ def add_datasample( classes) if gt_img_data is not None and pred_img_data is not None: - print(gt_img_data.dtype, pred_img_data.dtype) drawn_img = np.concatenate((gt_img_data, pred_img_data), axis=1) elif gt_img_data is not None: drawn_img = gt_img_data