def CreateDataLoader(_root,
_list_dir,
_input_height,
_input_width,
is_flip=True,
shuffle=True):
data_loader = None
from data.aligned_data_loader import AlignedDataLoader
data_loader = AlignedDataLoader(_root, _list_dir, _input_height,
_input_width, is_flip, shuffle)
return data_loader
def CreateDataLoader(opt):
data_loader = None
if opt.name == 'mrf':
from data.aligned_data_loader import AlignedNpDataLoader
data_loader = AlignedNpDataLoader()
elif opt.align_data:
from data.aligned_data_loader import AlignedDataLoader
data_loader = AlignedDataLoader()
else:
from data.unaligned_data_loader import UnalignedDataLoader
data_loader = UnalignedDataLoader()
print(data_loader.name())
data_loader.initialize(opt)
return data_loader
def CreateDataLoader(_root, _list_dir):
data_loader = None
from data.aligned_data_loader import AlignedDataLoader
data_loader = AlignedDataLoader(_root, _list_dir)
# if opt.align_data > 0:
# from data.aligned_data_loader import AlignedDataLoader
# data_loader = AlignedDataLoader()
# else:
# from data.unaligned_data_loader import UnalignedDataLoader
# data_loader = UnalignedDataLoader()
# print(data_loader.name())
# data_loader.initialize(opt)
return data_loader
def CreateDataLoader(opt):
data_loader = None
if opt.align_data == 1:
from data.aligned_data_loader import AlignedDataLoader
data_loader = AlignedDataLoader()
elif opt.align_data == 2:
from data.unaligned_data_loader import UnalignedDataLoader
data_loader = UnalignedDataLoader()
elif opt.align_data == 3:
from data.three_aligned_data_loader import ThreeAlignedDataLoader
data_loader = ThreeAlignedDataLoader()
elif opt.align_data == 4:
from data.four_aligned_data_loader import FourAlignedDataLoader
data_loader = FourAlignedDataLoader()
print(data_loader.name())
data_loader.initialize(opt)
return data_loader