🚀 Feature Checked the code and it doesn't seem possible to seed t

Hi <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="

Is it possible to seed augmentations about augly HOT 3 OPEN

facebookresearch commented on September 23, 2024

Is it possible to seed augmentations

from augly.

Comments (3)

zpapakipos commented on September 23, 2024 2

Hi @ierezell, thanks for this request! We currently have a seed arg for some augmentations that include random sampling within them (e.g. change_case in text https://github.com/facebookresearch/AugLy/blob/main/augly/text/functional.py#L65 or perspective_transform in image https://github.com/facebookresearch/AugLy/blob/main/augly/image/functional.py#L1719), but it's true that this is missing for some (e.g. insert_punctuation_chars in text https://github.com/facebookresearch/AugLy/blob/main/augly/text/functional.py#L144).

I will add this to our list of tasks to do soon, to add seed args to all augmentations that involve random sampling which don't already have it.

from augly.

sciencecw commented on September 23, 2024

It is good that change_case has seed argument, but the current implementation means that without specifying the seed, it keeps generating the same result

from augly.

javiabellan commented on September 23, 2024

AugLy is cool, however many transformations are not random:

Here is a code snippet to make any transfomation random, (even with the random distributions you preffer):

import numpy as np
import augly.image as imaugs
    
def random_uniform(min,max):
    return lambda: np.random.uniform(min, max)
    
def random_normal(mean,std):
    return lambda: np.random.normal(mean, std)

def randomize(fn_2_randomize, **kwargs):
    def randomized_fn(img_pil):
        # 1) Compute random_params into fixed_params:
        fixed_args = {}
        for key_paramName, value_randomFn in kwargs.items():
            fixed_args[key_paramName] = value_randomFn()
        # 2) Call PIL function with fixed_params
        return fn_2_randomize(img_pil, **fixed_args)
    return randomized_fn

random_overlay_text = randomize(imaugs.overlay_text,
                                x_pos=random_uniform(0,0.5),
                                y_pos=random_uniform(0,0.9))

# Now you can call it :)
random_overlay_text(img)