added embed for bugfix

2023-10-26 08:41:57 +02:00 · 2023-10-26 08:41:57 +02:00 · 68b3967a21
commit 68b3967a21
parent 7386bd83a5
1 changed files with 3 additions and 0 deletions
--- 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()):