new perlin

alexriedel1 · Dec 5, 2024 · 07a68ff · 07a68ff
1 parent f1a634e
commit 07a68ff
Show file tree

Hide file tree

Showing 9 changed files with 295 additions and 309 deletions.
diff --git a/src/anomalib/data/transforms/__init__.py b/src/anomalib/data/transforms/__init__.py
@@ -4,5 +4,6 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from .center_crop import ExportableCenterCrop
+from .multi_random_choice import MultiRandomChoice
 
-__all__ = ["ExportableCenterCrop"]
+__all__ = ["ExportableCenterCrop", "MultiRandomChoice"]
diff --git a/src/anomalib/data/utils/__init__.py b/src/anomalib/data/utils/__init__.py
@@ -3,10 +3,10 @@
 # Copyright (C) 2022-2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from .augmenter import Augmenter
 from .boxes import boxes_to_anomaly_maps, boxes_to_masks, masks_to_boxes
 from .download import DownloadInfo, download_and_extract
-from .generators import random_2d_perlin
+from .generators import generate_perlin_noise
+
 from .image import (
     generate_output_image_filename,
     get_image_filenames,
@@ -30,7 +30,7 @@
     "generate_output_image_filename",
     "get_image_filenames",
     "get_image_height_and_width",
-    "random_2d_perlin",
+    "generate_perlin_noise",
     "read_image",
     "read_mask",
     "read_depth_image",
@@ -42,7 +42,6 @@
     "TestSplitMode",
     "LabelName",
     "DirType",
-    "Augmenter",
     "masks_to_boxes",
     "boxes_to_masks",
     "boxes_to_anomaly_maps",

diff --git a/src/anomalib/data/utils/augmenter.py b/src/anomalib/data/utils/augmenter.py
diff --git a/src/anomalib/data/utils/generators/__init__.py b/src/anomalib/data/utils/generators/__init__.py
@@ -3,6 +3,6 @@
 # Copyright (C) 2022 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
-from .perlin import random_2d_perlin
+from .perlin import PerlinAnomalyGenerator, generate_perlin_noise
 
-__all__ = ["random_2d_perlin"]
+__all__ = ["PerlinAnomalyGenerator", "generate_perlin_noise"]