diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index 1794888d..d1930b9c 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license -__version__ = "8.3.61" +__version__ = "8.3.62" import os diff --git a/ultralytics/data/augment.py b/ultralytics/data/augment.py index 72152693..85b90148 100644 --- a/ultralytics/data/augment.py +++ b/ultralytics/data/augment.py @@ -1860,6 +1860,9 @@ class Albumentations: if self.contains_spatial else A.Compose(T) ) + if hasattr(self.transform, "set_random_seed"): + # Required for deterministic transforms in albumentations>=1.4.21 + self.transform.set_random_seed(torch.initial_seed()) LOGGER.info(prefix + ", ".join(f"{x}".replace("always_apply=False, ", "") for x in T if x.p)) except ImportError: # package not installed, skip pass