HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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
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