| |
| |
|
|
| import numpy as np |
|
|
| from util.logger import log |
| from lib.rays import rays_from_silh |
| from models.smpl import model_params_in_camera_coords |
|
|
| sess = None |
|
|
|
|
| class FrameData(object): |
| pass |
|
|
|
|
| def batch_invert(x): |
| try: |
| import tensorflow as tf |
| global sess |
|
|
| tx = tf.convert_to_tensor(x, dtype=tf.float32) |
| txi = tf.transpose(tf.matrix_inverse(tf.transpose(tx))) |
|
|
| if sess is None: |
| sess = tf.Session() |
|
|
| return sess.run(txi) |
|
|
| except ImportError: |
| log.info('Could not load tensorflow. Falling back to matrix inversion with numpy (slower).') |
|
|
| return np.asarray([np.linalg.inv(t) for t in x.T]).T |
|
|
|
|
| def setup_frame_rays(base_smpl, camera, camera_t, camera_rt, pose, trans, mask): |
| f = FrameData() |
|
|
| f.trans, f.pose = model_params_in_camera_coords(trans, pose, base_smpl.J[0], camera_t, camera_rt) |
| f.mask = mask |
|
|
| base_smpl.pose[:] = f.pose |
| camera.t[:] = f.trans |
|
|
| f.Vi = batch_invert(base_smpl.V.r) |
| f.rays = rays_from_silh(f.mask, camera) |
|
|
| return f |
|
|
