diff --git a/utils/visualizer.py b/utils/visualizer.py index b6799ed..4a1008b 100644 --- a/utils/visualizer.py +++ b/utils/visualizer.py @@ -886,6 +886,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 = np.clip(color, 0, 1) color = np.maximum(list(mplc.to_rgb(color)), 0.2) color[np.argmax(color)] = max(0.8, np.max(color)) @@ -1276,4 +1277,4 @@ def get_output(self): output (VisImage): the image output containing the visualizations added to the image. """ - return self.output \ No newline at end of file + return self.output