raab/efishSignalDetector
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

bf

Browse Source
This commit is contained in:
Till Raab 2023-10-26 08:49:40 +02:00
parent 10f8f872f6
commit efd860a127
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
inference.py
View File

@ -15,10 +15,6 @@ import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle from matplotlib.patches import Rectangle
def plot_inference(img_tensor, output, target, detection_threshold): def plot_inference(img_tensor, output, target, detection_threshold):
embed()
quit()
fig, ax = plt.subplots(figsize=(7, 7), num=target['image_id']) fig, ax = plt.subplots(figsize=(7, 7), num=target['image_id'])
ax.imshow(img_tensor.cpu().squeeze().permute(1, 2, 0), aspect='auto') 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()): for (x0, y0, x1, y1), l, score in zip(output['boxes'].cpu(), output['labels'].cpu(), output['scores'].cpu()):