Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Track Random Augmentations as They Are Applied #836

Open
aidansmyth95 opened this issue Jun 14, 2023 · 0 comments
Open

Track Random Augmentations as They Are Applied #836

aidansmyth95 opened this issue Jun 14, 2023 · 0 comments

Comments

@aidansmyth95
Copy link

I am working on a project where I am tracking the roll estimations for a face in an image. When I possibly rotate an image using a Sequence of probabilistic augmentations, I need to return the rotation that occurred (if any).

train_aug = Sequential(
[
Resize(tgt_img_size[:2], interpolation="linear"),
Fliplr(0.3),
Sometimes(0.3, Affine(rotate=10)),
]
)

Calling get_parameters() returns the distributions but not the parameters that were sampled from the distributions.

What I really need is something like:
{
flipped_lr: True,
rotate: [True, -4.5]
}

I would appreciate a code snippet demonstrating how people track imgaug operations performed like described above. I appreciate that this project is not being actively maintained, but I do believe that this already exists and I am just missing some documentation on it or a code example.

Thanks in advance.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant