Moved data augmentation

Dawars · Sep 18, 2023 · 6020b61 · 6020b61
1 parent 2a15725
commit 6020b61
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 3 deletions.
diff --git a/docmae/data/augmentation/__init__.py b/docmae/data/augmentation/__init__.py
diff --git a/docmae/utils/transforms.py → .../data/augmentation/random_resized_crop.py b/docmae/utils/transforms.py → .../data/augmentation/random_resized_crop.py
diff --git a/docmae/utils/replace_background.py → ...e/data/augmentation/replace_background.py b/docmae/utils/replace_background.py → ...e/data/augmentation/replace_background.py
diff --git a/docmae/fine_tune.py b/docmae/fine_tune.py
@@ -18,7 +18,7 @@
 from docmae.models.docmae import DocMAE
 
 from docmae import setup_logging
-from docmae.utils.transforms import RandomResizedCropWithUV
+from docmae.data.augmentation.random_resized_crop import RandomResizedCropWithUV
 
 logger = logging.getLogger(__name__)
 

diff --git a/tests/datasets.py b/tests/datasets.py
@@ -7,8 +7,8 @@
 
 from docmae.data.doc3d import Doc3D
 from docmae.data.docaligner import DocAligner
-from docmae.utils.replace_background import ReplaceBackground
-from docmae.utils.transforms import RandomResizedCropWithUV
+from docmae.data.augmentation.random_resized_crop import RandomResizedCropWithUV
+from docmae.data.augmentation.replace_background import ReplaceBackground
 
 os.environ["OPENCV_IO_ENABLE_OPENEXR"] = "1"