def __init__(self, opt):
BaseLoader.__init__(self, opt)
self.dirA = opt.dirA
self.dirB = opt.dirB
self.pathsA = file_utils.load_paths(self.dirA)
self.pathsB = file_utils.load_paths(self.dirB)
self.indexA = 0
self.indexB = 0
def __init__(self, opt):
BaseLoader.__init__(self, opt)
self.dirA = opt.dirA
self.dirB = opt.dirB
self.pathsA = file_utils.load_paths(self.dirA)
self.pathsB = file_utils.load_paths(self.dirB)
if len(self.pathsA) != len(self.pathsB):
print('Paired loader requires two datasets of equal length.')
sys.exit(0)
else:
self.paths = zip(self.pathsA, self.pathsB)
self.index = 0
def __init__(self, opt, training):
BaseDataset.__init__(self, opt, training)
self.dirA = opt.dirA
self.dirB = opt.dirB
self.pathsA = file_utils.load_paths(self.dirA)
self.pathsB = file_utils.load_paths(self.dirB)
def __init__(self, opt, training):
BaseDataset.__init__(self, opt, training)
self.dir = opt.dir
self.paths = file_utils.load_paths(self.dir)
def __init__(self, opt):
BaseLoader.__init__(self, opt)
self.dir = opt.dir
self.paths = file_utils.load_paths(self.dir)
self.index = 0