ultralytics 8.1.40 search in Python sets {} for speed (#9450)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-04-01 00:16:52 +02:00 · 2024-04-01 00:16:52 +02:00 · ea527507fe
commit ea527507fe
parent 30484d5925
41 changed files with 97 additions and 93 deletions
--- a/ultralytics/data/split_dota.py
+++ b/ultralytics/data/split_dota.py
@ -71,7 +71,7 @@ def load_yolo_dota(data_root, split="train"):
                    - train
                    - val
    """
-    assert split in ["train", "val"]
+    assert split in {"train", "val"}, f"Split must be 'train' or 'val', not {split}."
    im_dir = Path(data_root) / "images" / split
    assert im_dir.exists(), f"Can't find {im_dir}, please check your data root."
    im_files = glob(str(Path(data_root) / "images" / split / "*"))