From fba3eb7ca84999246806d218d3ffbdeb1c6264fc Mon Sep 17 00:00:00 2001 From: andres Date: Thu, 27 Jul 2023 16:03:03 +0200 Subject: [PATCH 1/2] fixing bugs for demo --- demo/demo.py | 2 +- demo/visualizer.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/demo/demo.py b/demo/demo.py index 3c4d0b0..7442348 100644 --- a/demo/demo.py +++ b/demo/demo.py @@ -124,7 +124,7 @@ def get_parser(): if len(args.input) == 1: for k in visualized_output.keys(): os.makedirs(k, exist_ok=True) - out_filename = os.path.join(k, args.output) + out_filename = os.path.join(args.output, k) visualized_output[k].save(out_filename) else: for k in visualized_output.keys(): diff --git a/demo/visualizer.py b/demo/visualizer.py index 5809bc8..9a5ca68 100644 --- a/demo/visualizer.py +++ b/demo/visualizer.py @@ -224,7 +224,7 @@ def non_empty_mask(self): assert ( len(empty_ids) == 1 ), ">1 ids corresponds to no labels. This is currently not supported" - return (self._seg != empty_ids[0]).numpy().astype(np.bool) + return (self._seg != empty_ids[0]).numpy().astype(bool) def semantic_masks(self): for sid in self._seg_ids: @@ -232,14 +232,14 @@ def semantic_masks(self): if sinfo is None or sinfo["isthing"]: # Some pixels (e.g. id 0 in PanopticFPN) have no instance or semantic predictions. continue - yield (self._seg == sid).numpy().astype(np.bool), sinfo + yield (self._seg == sid).numpy().astype(bool), sinfo def instance_masks(self): for sid in self._seg_ids: sinfo = self._sinfo.get(sid) if sinfo is None or not sinfo["isthing"]: continue - mask = (self._seg == sid).numpy().astype(np.bool) + mask = (self._seg == sid).numpy().astype(bool) if mask.sum() > 0: yield mask, sinfo From a82a08124ae9ffbb4cb450e1731d7440e8871199 Mon Sep 17 00:00:00 2001 From: andres Date: Thu, 27 Jul 2023 16:33:12 +0200 Subject: [PATCH 2/2] color min/max bug --- demo/visualizer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/demo/visualizer.py b/demo/visualizer.py index 9a5ca68..7706ad3 100644 --- a/demo/visualizer.py +++ b/demo/visualizer.py @@ -966,6 +966,7 @@ def draw_text( font_size = self._default_font_size # since the text background is dark, we don't want the text to be dark + color = tuple(np.clip(color, a_min=0, a_max=1)) color = np.maximum(list(mplc.to_rgb(color)), 0.2) color[np.argmax(color)] = max(0.8, np.max(color))