Applying different data augmentation per image in a mini-batch

Thanks, your solution works well!