xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 84b18f968cbb316787a5fd644efe7454ec50f411 into 453c6e38a51e9d1d5a2aa5fb7f1014a711913397

Browse Source
This commit is contained in:
Xiang Yang 2025-03-14 11:38:51 +08:00 committed by GitHub
commit 8900452ca2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/models/yolo/detect/val.py
View File

@ -65,7 +65,7 @@ class DetectionValidator(BaseValidator):
def init_metrics(self, model): def init_metrics(self, model):
"""Initialize evaluation metrics for YOLO.""" """Initialize evaluation metrics for YOLO."""
val = self.data.get(self.args.split, "") # validation path 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.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.args.save_json |= self.is_coco # run on final val if training COCO
self.names = model.names self.names = model.names