iPhone rectified centertest routines added. [moberweger#1]

mahdinobar · Apr 4, 2020 · d10415c · d10415c
1 parent 2a18840
commit d10415c
Show file tree

Hide file tree

Showing 4 changed files with 704 additions and 10 deletions.
diff --git a/src/data/dataset.py b/src/data/dataset.py
@@ -23,7 +23,7 @@
 """
 
 import numpy
-from data.importers import NYUImporter, ICVLImporter, MSRA15Importer
+from data.importers import NYUImporter, ICVLImporter, MSRA15Importer, iPhoneImporter
 
 
 __author__ = "Paul Wohlhart <wohlhart@icg.tugraz.at>, Markus Oberweger <oberweger@icg.tugraz.at>"
@@ -123,6 +123,18 @@ def __init__(self, imgSeqs=None, basepath=None, localCache=True):
         self.lmi = ICVLImporter(basepath)
 
 
+class iPhoneDataset(Dataset):
+    def __init__(self, imgSeqs=None, basepath=None, localCache=True):
+        """
+        constructor
+        """
+        super(iPhoneDataset, self).__init__(imgSeqs, localCache)
+        if basepath is None:
+            basepath = '../../data/iPhone/'
+
+        self.lmi = iPhoneImporter(basepath)
+
+
 class MSRA15Dataset(Dataset):
     def __init__(self, imgSeqs=None, basepath=None, localCache=True):
         """