ultralytics 8.0.58 new SimpleClass, fixes and updates (#1636)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2023-03-26 22:16:38 +02:00 · 2023-03-26 22:16:38 +02:00 · ec10002a4a
commit ec10002a4a
parent ef03e6732a
30 changed files with 351 additions and 314 deletions
--- a/ultralytics/yolo/utils/plotting.py
+++ b/ultralytics/yolo/utils/plotting.py
@ -295,7 +295,7 @@ def plot_images(images,
            for j, box in enumerate(boxes.T.tolist()):
                c = classes[j]
                color = colors(c)
-                c = names[c] if names else c
+                c = names.get(c, c) if names else c
                if labels or conf[j] > 0.25:  # 0.25 conf thresh
                    label = f'{c}' if labels else f'{c} {conf[j]:.1f}'
                    annotator.box_label(box, label, color=color)