fix: Deci-AI#2047

Save target map for ImagesDetectionPrediction
fitagdinov · Sep 15, 2024 · 947fc84 · 947fc84
1 parent 4797c97
commit 947fc84
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/super_gradients/training/utils/predict/prediction_results.py b/src/super_gradients/training/utils/predict/prediction_results.py
@@ -554,6 +554,9 @@ def save(
                 box_thickness=box_thickness,
                 show_confidence=show_confidence,
                 color_mapping=color_mapping,
+                target_bboxes=target_bbox,
+                target_bboxes_format=target_bboxes_format,
+                target_class_ids=target_class_id,
                 class_names=class_names,
             )