From 84b18f968cbb316787a5fd644efe7454ec50f411 Mon Sep 17 00:00:00 2001
From: zhiqiang yang <89863442+yang-0201@users.noreply.github.com>
Date: Thu, 18 Jul 2024 10:39:57 +0800
Subject: [PATCH] Update val.py
Fixed a bug in yolov10's verification of the MAP of the COCO test2017
---
ultralytics/models/yolo/detect/val.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/ultralytics/models/yolo/detect/val.py b/ultralytics/models/yolo/detect/val.py
index 5550ec3e..6650046a 100644
--- a/ultralytics/models/yolo/detect/val.py
+++ b/ultralytics/models/yolo/detect/val.py
@@ -65,7 +65,7 @@ class DetectionValidator(BaseValidator):
def init_metrics(self, model):
"""Initialize evaluation metrics for YOLO."""
val = self.data.get(self.args.split, "") # validation path
- self.is_coco = isinstance(val, str) and "coco" in val and val.endswith(f"{os.sep}val2017.txt") # is COCO
+ self.is_coco = isinstance(val, str) and "coco" in val and (val.endswith(f"{os.sep}val2017.txt") or val.endswith(f"{os.sep}test-dev2017.txt")) # is COCO
self.class_map = converter.coco80_to_coco91_class() if self.is_coco else list(range(1000))
self.args.save_json |= self.is_coco # run on final val if training COCO
self.names = model.names