diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-18 03:38:47 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-18 03:38:47 -0700 |
| commit | c99ce7c4e781712e0252c6127ad1a4e8021cc489 (patch) | |
| tree | ba99dfd56a47036d9c1f18620abf4efc248839ab /data/data_loader.py | |
first commit
Diffstat (limited to 'data/data_loader.py')
| -rw-r--r-- | data/data_loader.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/data/data_loader.py b/data/data_loader.py new file mode 100644 index 0000000..69035ea --- /dev/null +++ b/data/data_loader.py @@ -0,0 +1,12 @@ + +def CreateDataLoader(opt): + data_loader = None + 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 |