Skip to content

Instantly share code, notes, and snippets.

@thigm85
Created November 12, 2021 15:51
Show Gist options
  • Save thigm85/3cd7d865e0d3c906a81c1311c407577c to your computer and use it in GitHub Desktop.
Save thigm85/3cd7d865e0d3c906a81c1311c407577c to your computer and use it in GitHub Desktop.
image_dataset = ImageDataset(
img_dir=os.environ["IMG_DIR"],
transform=preprocess
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment