From efd860a12747c816c68835aa09370a1156c2aa08 Mon Sep 17 00:00:00 2001 From: Till Raab Date: Thu, 26 Oct 2023 08:49:40 +0200 Subject: [PATCH] bf --- inference.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/inference.py b/inference.py index 505d313..47bb68c 100644 --- a/inference.py +++ b/inference.py @@ -15,10 +15,6 @@ import matplotlib.pyplot as plt 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.cpu().squeeze().permute(1, 2, 0), aspect='auto') for (x0, y0, x1, y1), l, score in zip(output['boxes'].cpu(), output['labels'].cpu(), output['scores'].cpu()):