Thanks for your reply. In offline augmentation, I applied rotation and zoom to the images. And in online augmentation I applied randomcrop, flip and resize etc. as shown in the above code snippet. Would it work?