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

preicted boxes in dashed green

Browse Source
This commit is contained in:
Till Raab 2023-10-26 09:05:08 +02:00
parent 4373b7c453
commit 72c58af7e4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference.py
View File

@ -26,7 +26,7 @@ def plot_inference(img_tensor, output, target, detection_threshold):
Rectangle((x0, y0), Rectangle((x0, y0),
(x1 - x0), (x1 - x0),
(y1 - y0), (y1 - y0),
fill=False, color="tab:green", linewidth=2, zorder=10) fill=False, color="tab:green", linestyle='--', linewidth=2, zorder=10)
) )
for (x0, y0, x1, y1), l in zip(target['boxes'], target['labels']): for (x0, y0, x1, y1), l in zip(target['boxes'], target['labels']):
ax.add_patch( ax.add_patch(