Imported from https://github.com/junkwhinger/fastautoaugment_jsh