Imported from https://github.com/JunHahn/image-augmentation-workspace