applying custom augmentations

luxonis · Dec 14, 2024 · 90197f9 · 90197f9
1 parent 6f840f2
commit 90197f9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/luxonis_ml/data/augmentations/albumentations_engine.py b/luxonis_ml/data/augmentations/albumentations_engine.py
@@ -409,6 +409,7 @@ def apply(self, input_batch: List[LoaderOutput]) -> LoaderOutput:
                 del data[target_name]
 
         data = self.spatial_transform(**data)
+        data = self.custom_transform(**data)
 
         transformed_size = data["image"].shape[:2]