save

utils/data_utils.py

@@ -63,9 +63,7 @@ def center_crop_resize(img, new_center, size):
 
 
 """ CAMERA """
-def intrinsic():
-	RES_X = 640
-	RES_Y = 480
+def intrinsic(RES_X=640, RES_Y=480):
 	f_mm             = 50 # blender default
 	sensor_w_mm      = 36 # blender default
 	sensor_h_mm = sensor_w_mm * RES_Y / RES_X
@@ -108,3 +106,16 @@ def get_camera(cam, img_size, device):
     K = torch.tensor(intrinsic()[None, :, :], device=device)
     cameras = cameras_from_opencv_projection(R=ext[None, :, :3], tvec=ext[None, :, 3], camera_matrix=K, image_size=torch.tensor(size[None, :], device=device))
     return cameras, K
+
+
+def filter_info(info, frame_nr):
+    frame_specific = set(['poses', 'trans'])
+    general = set(['shape', 'camLoc', 'zrot', 'gender'])
+    out = dict()
+    for key in info.keys():
+        if key in frame_specific:
+            out[key] = info[key][:, frame_nr]
+        elif key in general:
+            out[key] = info[key]
+    
+    return out