Skip to content

Instantly share code, notes, and snippets.

@JoshVarty
Created January 29, 2019 00:08
Show Gist options
  • Save JoshVarty/697876a0ddc82691dd786b2d4169ba8b to your computer and use it in GitHub Desktop.
Save JoshVarty/697876a0ddc82691dd786b2d4169ba8b to your computer and use it in GitHub Desktop.
path = 'data/counting'
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)
data.show_batch(rows=3, figsize=(7,8))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment