Skip to content

Instantly share code, notes, and snippets.

@Yuktha-Majella

Yuktha-Majella/Data_augmentation

Last active August 10, 2021 16:26
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Yuktha-Majella/ca397a4a4937489f920c04a186f28b36 to your computer and use it in GitHub Desktop.
Save Yuktha-Majella/ca397a4a4937489f920c04a186f28b36 to your computer and use it in GitHub Desktop.
Download ZIP
Data Augmentation using ImageDataBunch to increase the size of Training set
Raw
Data_augmentation
tfms = get_transforms(do_flip=True, flip_vert=True, max_rotate=50, max_lighting=0.1, max_warp=0 )
data = ImageDataBunch.from_df('/content/drive/MyDrive/CV_Vehicle_classification/train_data/images', train, ds_tfms=tfms, label_delim= None, valid_pct=0.2, fn_col=0, label_col=1 , size=299,bs=64).normalize(imagenet_stats)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment