That would work using my approach, but as I said, you would stick to these “static” transformations/augmentations.
I’m not really that familiar with Colab, but wouldn’t it be possible to download and unzipping the dataset once and store it in your Google Drive?