Get image size from settings

data/create_syn_data.py

@@ -143,8 +143,8 @@ def create_data(out_root, idx, n_samples, imsize, patterns, K, baseline, blend_i
             fy = K[1, 1] * scale
             px = K[0, 2] * scale
             py = K[1, 2] * scale
-            im_height = imsize[0] * scale
+            im_height = imsize[0] * scale // 1
-            im_width = imsize[1] * scale
+            im_width = imsize[1] * scale // 1
             cams.append(renderer.PyCamera(fx, fy, px, py, Rcam, tcam, im_width, im_height))
             projs.append(renderer.PyCamera(fx, fy, px, py, Rproj, tproj, im_width, im_height))
 
@@ -221,6 +221,7 @@ if __name__ == '__main__':
         config = json.load(fp)
         data_root = Path(config['DATA_ROOT'])
         shapenet_root = config['SHAPENET_ROOT']
+        settings_imsize = tuple(map(int, config['IMSIZE'].split(',')))
 
     data_type = 'syn'
     out_root = data_root / f'{data_type}'
@@ -242,6 +243,9 @@ if __name__ == '__main__':
 
     # camera parameters
     imsize, K = load_camera_parameters()
+    # remember what was set in the settings
+    imsize = settings_imsize
+
     imsizes = [(imsize[0] // (2 ** s), imsize[1] // (2 ** s)) for s in range(4)]
 
     focal_lengths = [K[0, 0] / (2 ** s) for s in range(4)]