From 68b3967a21c67d71757f4f655480838a8a9d43ba Mon Sep 17 00:00:00 2001
From: Till Raab <till.raab@uni-tuebingen.de>
Date: Thu, 26 Oct 2023 08:41:57 +0200
Subject: [PATCH] added embed for bugfix
---
inference.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/inference.py b/inference.py
index 4351ed4..687c53f 100644
--- a/inference.py
+++ b/inference.py
@@ -16,6 +16,9 @@ from matplotlib.patches import Rectangle
def plot_inference(img_tensor, output, target, detection_threshold):
+ embed()
+ quit()
+
fig, ax = plt.subplots(figsize=(7, 7), num=target['image_id'])
ax.imshow(img_tensor.squeeze().permute(1, 2, 0), aspect='auto')
for (x0, y0, x1, y1), l, score in zip(output[0]['boxes'].cpu(), output[0]['labels'].cpu(), output[0]['scores'].cpu()):