I need to add an augmentation rate, which indicates the probability to do the style transfer for a give input.
p1 = 1
p2 = 0.5
x = random.random()
y = random.random()
if x <= p1:
do allow augmentation for this samples
if y < p2:
do augmentation for this slice of image

I need to add an augmentation rate, which indicates the probability to do the style transfer for a give input.
p1 = 1
p2 = 0.5
x = random.random()
y = random.random()
if x <= p1:

do allow augmentation for this samples
if y < p2:
do augmentation for this slice of image