diffusion-models-class
diffusion-models-class copied to clipboard
add code to load images from local folder
Added code example for loading images from a local folder.
Check out this pull request on ![]()
See visual diffs & provide feedback on Jupyter Notebooks.
Powered by ReviewNB
@daspartho good idea. To elaborate (especially for colab users) we can include:
from google.colab import drive
drive.mount('/content/gdrive')
PATH= "/content/gdrive/MyDrive/<FOLDER>/" //pytorch will expect <train> subfolder inside
!ls {PATH}
dataset = torchvision.datasets.ImageFolder(root=PATH, transform=
torchvision.transforms.Compose([
torchvision.transforms.ToTensor(),
torchvision.transforms.Resize(28),
]))