MCPcopy Create free account
hub / github.com/HobbitLong/PyContrast / __init__

Method __init__

pycontrast/datasets/dataset.py:11–17  ·  view source on GitHub ↗
(self, root, transform=None, target_transform=None,
                 two_crop=False, jigsaw_transform=None)

Source from the content-addressed store, hash-verified

9 """Folder datasets which returns the index of the image (for memory_bank)
10 """
11 def __init__(self, root, transform=None, target_transform=None,
12 two_crop=False, jigsaw_transform=None):
13 super(ImageFolderInstance, self).__init__(root, transform, target_transform)
14 self.two_crop = two_crop
15 self.jigsaw_transform = jigsaw_transform
16 self.use_jigsaw = (jigsaw_transform is not None)
17 self.num = self.__len__()
18
19 def __getitem__(self, index):
20 """

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected