update dataset

yxgeee · Aug 14, 2020 · f5de741 · f5de741
1 parent c2a0d79
commit f5de741
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ibl/datasets/tokyo.py b/ibl/datasets/tokyo.py
@@ -37,9 +37,9 @@ def arrange(self):
         raw_dir = osp.join(self.root, 'raw')
         if (not osp.isdir(raw_dir)):
             raise RuntimeError("Dataset not found.")
-        TM_root = osp.join('TokyoTM', 'images')
-        db_root = osp.join('Tokyo247', 'images')
-        q_root = osp.join('Tokyo247', 'query')
+        TM_root = osp.join('tokyoTM', 'images')
+        db_root = osp.join('tokyo247', 'images')
+        q_root = osp.join('tokyo247', 'query')
 
         identities = []
         utms = []
@@ -115,6 +115,7 @@ def register_247(split):
             db_ids = []
             for fpath, utm in zip(struct.dbImage, struct.utmDb):
                 sid = osp.dirname(fpath)
+                fpath = fpath[:-3]+'png'
                 # sid = str(utm[0])+'_'+str(utm[1])
                 # sid = fpath[:-12]
                 if (sid not in db_pids.keys()):