diff --git a/mmdet/datasets/transforms/transforms.py b/mmdet/datasets/transforms/transforms.py index c50b987db33..7abcf5ec9cd 100644 --- a/mmdet/datasets/transforms/transforms.py +++ b/mmdet/datasets/transforms/transforms.py @@ -32,9 +32,14 @@ try: import albumentations from albumentations import Compose + + # This assumes albumentations' versions are just ints separated by + # periods, e.g. 1.4.11, allowing us to avoid a new dependency. + ALBU_VERSION = tuple(map(int, albumentations.__version__.split('.'))) except ImportError: albumentations = None Compose = None + ALBU_VERSION = None Number = Union[int, float] @@ -1627,8 +1632,13 @@ def __init__(self, self.bbox_params = ( self.albu_builder(bbox_params) if bbox_params else None) - self.aug = Compose([self.albu_builder(t) for t in self.transforms], - bbox_params=self.bbox_params) + if ALBU_VERSION is not None and ALBU_VERSION >= (1, 4, 11): + self.aug = Compose([self.albu_builder(t) for t in self.transforms], + bbox_params=self.bbox_params, + strict=False) + else: + self.aug = Compose([self.albu_builder(t) for t in self.transforms], + bbox_params=self.bbox_params) if not keymap: self.keymap_to_albu = {