CoreML NMS and half fixes (#143)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-01-05 00:20:54 +01:00
parent 55bdca6768
commit 172cef2d20
18 changed files with 95 additions and 414 deletions
--- a/ultralytics/yolo/data/utils.py
+++ b/ultralytics/yolo/data/utils.py
@ -201,7 +201,7 @@ def check_dataset_yaml(data, autodownload=True):
        extract_dir, autodownload = data.parent, False
    # Read yaml (optional)
    if isinstance(data, (str, Path)):
-        data = yaml_load(data)  # dictionary
+        data = yaml_load(data, append_filename=True)  # dictionary

    # Checks
    for k in 'train', 'val', 'names':