diff --git a/bitmind/__init__.py b/bitmind/__init__.py index 4d9770b..da53456 100644 --- 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 index e32d3c4..7accca7 100644 --- 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.