Draw most likely bbox on top (Deci-AI#1061) (Deci-AI#1063)

Co-authored-by: Shay Aharon <[email protected]>
Co-authored-by: Eugene Khvedchenya <[email protected]>

Author: 3 people

src/super_gradients/training/models/prediction_results.py

@@ -66,7 +66,7 @@ def draw(self, box_thickness: int = 2, show_confidence: bool = True, color_mappi
         image = self.image.copy()
         color_mapping = color_mapping or generate_color_mapping(len(self.class_names))
 
-        for pred_i in range(len(self.prediction)):
+        for pred_i in np.argsort(self.prediction.confidence):
 
             class_id = int(self.prediction.labels[pred_i])
             score = "" if not show_confidence else str(round(self.prediction.confidence[pred_i], 2))