diffusion-models-class icon indicating copy to clipboard operation
diffusion-models-class copied to clipboard

add code to load images from local folder

Open daspartho opened this issue 3 years ago • 2 comments

Added code example for loading images from a local folder.

daspartho avatar Dec 03 '22 18:12 daspartho

Check out this pull request on  ReviewNB

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),
                               ]))

RELNO avatar Dec 03 '22 21:12 RELNO