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

OBB: Fix when training on zip data (#8680)

Browse Source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Laughing 2024-03-06 01:37:33 +08:00 committed by GitHub
parent a7cfd83c5f
commit ddc94a6981
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ultralytics/engine/trainer.py
View File

@ -129,7 +129,12 @@ class BaseTrainer:
try: try:
if self.args.task == "classify": if self.args.task == "classify":
self.data = check_cls_dataset(self.args.data) self.data = check_cls_dataset(self.args.data)
elif self.args.data.split(".")[-1] in ("yaml", "yml") or self.args.task in ("detect", "segment", "pose"): elif self.args.data.split(".")[-1] in ("yaml", "yml") or self.args.task in (
"detect",
"segment",
"pose",
"obb",
):
self.data = check_det_dataset(self.args.data) self.data = check_det_dataset(self.args.data)
if "yaml_file" in self.data: if "yaml_file" in self.data:
self.args.data = self.data["yaml_file"] # for validating 'yolo train data=url.zip' usage self.args.data = self.data["yaml_file"] # for validating 'yolo train data=url.zip' usage