summaryrefslogtreecommitdiff
path: root/data/data_loader.py
diff options
context:
space:
mode:
Diffstat (limited to 'data/data_loader.py')
-rwxr-xr-xdata/data_loader.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/data/data_loader.py b/data/data_loader.py
new file mode 100755
index 0000000..2a4433a
--- /dev/null
+++ b/data/data_loader.py
@@ -0,0 +1,7 @@
+
+def CreateDataLoader(opt):
+ from data.custom_dataset_data_loader import CustomDatasetDataLoader
+ data_loader = CustomDatasetDataLoader()
+ print(data_loader.name())
+ data_loader.initialize(opt)
+ return data_loader