increasing frequency of more difficult augmentations (#146)

BitMind-AI · Jan 31, 2025 · 04a053b · 04a053b
1 parent 5d178f8
commit 04a053b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bitmind/__init__.py b/bitmind/__init__.py
@@ -18,7 +18,7 @@
 # DEALINGS IN THE SOFTWARE.
 
 
-__version__ = "2.1.2"
+__version__ = "2.1.3"
 version_split = __version__.split(".")
 __spec_version__ = (
     (1000 * int(version_split[0]))

diff --git a/bitmind/utils/image_transforms.py b/bitmind/utils/image_transforms.py
@@ -410,9 +410,9 @@ def get_random_augmentations_hard(target_image_size=TARGET_IMAGE_SIZE):
 
 def apply_augmentation_by_level(image, target_image_size, level_probs={
         0: 0.25,  # No augmentations (base transforms)
-        1: 0.45,  # Basic augmentations
-        2: 0.15,  # Medium distortions
-        3: 0.15   # Hard distortions
+        1: 0.25,  # Basic augmentations
+        2: 0.25,  # Medium distortions
+        3: 0.25   # Hard distortions
     }):
     """
     Apply image transformations based on randomly selected level.