MCPcopy
hub / github.com/NVlabs/SPADE / __init__

Method __init__

data/image_folder.py:73–85  ·  view source on GitHub ↗
(self, root, transform=None, return_paths=False,
                 loader=default_loader)

Source from the content-addressed store, hash-verified

71class ImageFolder(data.Dataset):
72
73 def __init__(self, root, transform=None, return_paths=False,
74 loader=default_loader):
75 imgs = make_dataset(root)
76 if len(imgs) == 0:
77 raise(RuntimeError("Found 0 images in: " + root + "\n"
78 "Supported image extensions are: " +
79 ",".join(IMG_EXTENSIONS)))
80
81 self.root = root
82 self.imgs = imgs
83 self.transform = transform
84 self.return_paths = return_paths
85 self.loader = loader
86
87 def __getitem__(self, index):
88 path = self.imgs[index]

Callers

nothing calls this directly

Calls 1

make_datasetFunction · 0.85

Tested by

no test coverage detected