Skip to content

Instantly share code, notes, and snippets.

@thecaffeinedev

thecaffeinedev/load_image_data.py Secret

Created December 28, 2019 11:09
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 thecaffeinedev/66c2fdbb4669da4540fba144604884cc to your computer and use it in GitHub Desktop.
Save thecaffeinedev/66c2fdbb4669da4540fba144604884cc to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_image_data.py
np.random.seed(42)
data = ImageDataBunch.from_folder(path, train=".", valid_pct=0.2,
ds_tfms=get_transforms(), size=224, num_workers=4).normalize(imagenet_stats)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment