save

2023-01-23 10:17:53 +00:00
parent f4341c5e6c
commit ab78dab165
4 changed files with 112 additions and 16 deletions
@@ -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