[Fix] Fix browse dataset (#9401)

* [Fix] Fix browse dataset. * clean code
open-mmlab · Nov 29, 2022 · b30c875 · b30c875
1 parent 59f35db
commit b30c875
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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)
 

diff --git 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)