From 72c58af7e4cd69ee7da252583c19be0e12987bdd Mon Sep 17 00:00:00 2001 From: Till Raab Date: Thu, 26 Oct 2023 09:05:08 +0200 Subject: [PATCH] preicted boxes in dashed green --- inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference.py b/inference.py index 27428b1..88b5357 100644 --- a/inference.py +++ b/inference.py @@ -26,7 +26,7 @@ def plot_inference(img_tensor, output, target, detection_threshold): Rectangle((x0, y0), (x1 - x0), (y1 - y0), - fill=False, color="tab:green", linewidth=2, zorder=10) + fill=False, color="tab:green", linestyle='--', linewidth=2, zorder=10) ) for (x0, y0, x1, y1), l in zip(target['boxes'], target['labels']): ax.add_patch(