Tracking Augmentation Parameters Applied

[aidansmyth95]

Hi there. I am looking for a way of tracking the augmentations that happen as I augment my batch of images one by one in a for loop.

My augmentation sequence looks like this:

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

I then apply it as follows:
aug_img = train_aug.augment_image(img)

It would be nice if I were able post-augmentation to determine what level of rotation and whether or not it was flipped.

Printing the augmentor parameters looks something like this:
[DiscreteUniform(Deterministic(int 224), Deterministic(int 224)), Deterministic(linear), 'HW']
[Binomial(Deterministic(float 0.30000000))]
[Binomial(Deterministic(float 0.30000000))]

Unfortunately, the above does not tell me how the input image was affected. Your help in this would be greatly appreciated.

Thanks!