update get_yolo_box(). if multiple objects have the same label, choose the one with the highest score. show the score on the overlay.

Author: jw098

SerialPrograms/Source/PokemonSV/Programs/AutoStory/PokemonSV_AutoStoryTools.cpp

@@ -1451,12 +1451,25 @@ ImageFloatBox get_yolo_box(
     yolo_detector.detect(snapshot);
 
     ImageFloatBox target_box{-1, -1, -1, -1};
+    double target_score = 0;
     for (YOLOv5Session::DetectionBox detected_box : detected_boxes){
         ImageFloatBox box = detected_box.box;
         std::string label = yolo_detector.session()->label_name(detected_box.label_idx);
+        double score = detected_box.score;
+        std::string score_string = std::to_string(std::round(score * 100.0) / 100.0);
+        size_t end = score_string.find_last_not_of('0');
+        if (end != std::string::npos){
+            score_string.erase(end + 1);  // remove trailing zeros
+        }else{
+            score_string.clear();
+        }
+        std::string label_score = label + ": " + score_string;
         if (target_label == label){
-            target_box = box;
-            overlays.add(COLOR_RED, box, label);
+            overlays.add(COLOR_RED, box, label_score);
+            
+            if (score > target_score){
+                target_box = box;
+            }
         }else{
             overlays.add(COLOR_BLUE, box, label);
         }

SerialPrograms/Source/PokemonSV/Programs/AutoStory/PokemonSV_AutoStoryTools.h

@@ -390,6 +390,7 @@ void move_player_forward(
 );
 
 // get the box of the target object
+// if multiple objects have the same label, choose the one with the highest score
 // return ImageFloatBox{-1, -1, -1, -1} if target object not found
 ImageFloatBox get_yolo_box(
     SingleSwitchProgramEnvironment& env,