Update plots.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-21 15:57:51 +01:00 · 2024-11-21 15:57:51 +01:00 · 2101bcf12c
parent f255c7ae06
commit 2101bcf12c
1 changed files with 1 additions and 1 deletions
--- a/utils/plots.py
+++ b/utils/plots.py
@ -183,7 +183,7 @@ def plot_images(images, targets, paths=None, fname="images.jpg", names=None):
    # Annotate
    fs = int((h + w) * ns * 0.01)  # font size
    annotator = Annotator(mosaic, line_width=round(fs / 10), font_size=fs, pil=True, example=names)
-    for i in range(bs):
+    for i in range(min(bs, i+1)):
        x, y = int(w * (i // ns)), int(h * (i % ns))  # block origin
        annotator.rectangle([x, y, x + w, y + h], None, (255, 255, 255), width=2)  # borders
        if paths: