update proc to include __imagekey__

hoho/hoho.py

@@ -158,31 +158,33 @@ def temp_working_directory():
 ############# Dataset ############# 
 def proc(row, split='train'):
     out = {}
+    out['__key__'] = None
+    out['__imagekey__'] = []
     for k, v in row.items():
-        colname = k.split('.')[0]
+        key_parts = k.split('.')
+        colname = key_parts[0]
+        if colname == 'ade20k': 
+            out['__imagekey__'].append(key_parts[1])
         if colname in {'ade20k', 'depthcm', 'gestalt'}:
             if colname in out:
                 out[colname].append(v)
             else:
                 out[colname] = [v]
         elif colname in {'wireframe', 'mesh'}:
-            # out.update({a: b.tolist() for a,b in v.items()})
             out.update({a: b for a,b in v.items()})
         elif colname in 'kr':
             out[colname.upper()] = v
         else:
             out[colname] = v
-            
     return Sample(out)
 
 
 
 
 
+
 from . import read_write_colmap
 def decode_colmap(s):
-    # with open('colmap_solve/points3D.bin', 'wb') as stream:
-    
     with temp_working_directory():
 
         with open('points3D.bin', 'wb') as stream:

setup.py

@@ -7,7 +7,7 @@ import glob
 required = []
 
 setup(name='hoho',
-	version='0.0.1',
+	version='0.0.2',
 	description='Tools and utilites for the HoHo Dataset and S23DR Competition',
 	url='usm3d.github.io',
 	author='Jack Langerman, Dmytro Mishkin, S23DR Orgainizing Team',