ultralytics 8.0.148 fix SettingsManager empty YAML bug (#4180)

This commit is contained in:
Glenn Jocher
2023-08-05 16:13:29 +02:00
committed by GitHub
parent a9fb8b239b
commit 038558cfab
7 changed files with 50 additions and 38 deletions

View File

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = '8.0.147'
__version__ = '8.0.148'
from ultralytics.hub import start
from ultralytics.models import RTDETR, SAM, YOLO

View File

@ -329,7 +329,10 @@ def yaml_load(file='data.yaml', append_filename=False):
s = re.sub(r'[^\x09\x0A\x0D\x20-\x7E\x85\xA0-\uD7FF\uE000-\uFFFD\U00010000-\U0010ffff]+', '', s)
# Add YAML filename to dict and return
return {**yaml.safe_load(s), 'yaml_file': str(file)} if append_filename else yaml.safe_load(s)
data = yaml.safe_load(s) or {} # always return a dict (yaml.safe_load() may return None for empty files)
if append_filename:
data['yaml_file'] = str(file)
return data
def yaml_print(yaml_file: Union[str, Path, dict]) -> None: