From 704a3c9d0cb124ff01e300ac13883b09c8d3d4f9 Mon Sep 17 00:00:00 2001 From: bjzhb666 Date: Tue, 23 Apr 2024 18:19:51 +0800 Subject: [PATCH] [Feature] add test without labels features --- mmseg/visualization/local_visualizer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmseg/visualization/local_visualizer.py b/mmseg/visualization/local_visualizer.py index ee3d652c7b..7e612b5092 100644 --- a/mmseg/visualization/local_visualizer.py +++ b/mmseg/visualization/local_visualizer.py @@ -103,7 +103,7 @@ def _draw_sem_seg(self, sem_seg: PixelData, classes: Optional[List], palette: Optional[List], - with_labels: Optional[bool] = True) -> np.ndarray: + with_labels: Optional[bool] = False) -> np.ndarray: """Draw semantic seg of GT or prediction. Args: @@ -265,7 +265,7 @@ def add_datasample( # TODO: Supported in mmengine's Viusalizer. out_file: Optional[str] = None, step: int = 0, - with_labels: Optional[bool] = True) -> None: + with_labels: Optional[bool] = False) -> None: """Draw datasample and save to all backends. - If GT and prediction are plotted at the same time, they are