Skip to content

Instantly share code, notes, and snippets.

@jacKlinc

jacKlinc/data_block_example.py

Created December 8, 2020 17:00
Show Gist options
  • Save jacKlinc/53e575629b5e2f99d5e567500a9cf1c9 to your computer and use it in GitHub Desktop.
Save jacKlinc/53e575629b5e2f99d5e567500a9cf1c9 to your computer and use it in GitHub Desktop.
Download ZIP
FastAI Data Block Creation
Raw
data_block_example.py
masks = DataBlock(
blocks=(ImageBlock, CategoryBlock),
get_items=get_image_files(path),
splitter=RandomSplitter(valid_pct=0.2, seed=42),
get_y=parent_label,
item_tfms=Resize(128))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment