Suggest largest when choose ap50 best conf

yolov6/core/evaler.py

@@ -202,7 +202,7 @@ def predict_model(self, model, dataloader, task):
 
                 from yolov6.utils.metrics import ap_per_class
                 p, r, ap, f1, ap_class = ap_per_class(*stats, plot=self.plot_curve, save_dir=self.save_dir, names=model.names)
-                AP50_F1_max_idx = f1.mean(0).argmax()
+                AP50_F1_max_idx = len(f1.mean(0)) - f1.mean(0)[::-1].argmax() -1
                 LOGGER.info(f"IOU 50 best mF1 thershold near {AP50_F1_max_idx/1000.0}.")
                 ap50, ap = ap[:, 0], ap.mean(1)  # [email protected], [email protected]:0.95
                 mp, mr, map50, map = p[:, AP50_F1_max_idx].mean(), r[:, AP50_F1_max_idx].mean(), ap50.mean(), ap.mean()