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()):