Update app.py

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import tensorflow as tf
 import numpy as np
 
+from transformations import *
+
 # Setting random seed to obtain reproducible results.
 tf.random.set_seed(42)
 
@@ -189,51 +191,6 @@ def render_rgb_depth(model, rays_flat, t_vals, rand=True, train=True):
         depth_map = tf.reduce_sum(weights * t_vals[:, None, None], axis=-1)
     return (rgb, depth_map)
 
-def get_translation_t(t):
-    """Get the translation matrix for movement in t."""
-    matrix = [
-        [1, 0, 0, 0],
-        [0, 1, 0, 0],
-        [0, 0, 1, t],
-        [0, 0, 0, 1],
-    ]
-    return tf.convert_to_tensor(matrix, dtype=tf.float32)
-
-
-def get_rotation_phi(phi):
-    """Get the rotation matrix for movement in phi."""
-    matrix = [
-        [1, 0, 0, 0],
-        [0, tf.cos(phi), -tf.sin(phi), 0],
-        [0, tf.sin(phi), tf.cos(phi), 0],
-        [0, 0, 0, 1],
-    ]
-    return tf.convert_to_tensor(matrix, dtype=tf.float32)
-
-
-def get_rotation_theta(theta):
-    """Get the rotation matrix for movement in theta."""
-    matrix = [
-        [tf.cos(theta), 0, -tf.sin(theta), 0],
-        [0, 1, 0, 0],
-        [tf.sin(theta), 0, tf.cos(theta), 0],
-        [0, 0, 0, 1],
-    ]
-    return tf.convert_to_tensor(matrix, dtype=tf.float32)
-
-
-def pose_spherical(theta, phi, t):
-    """
-    Get the camera to world matrix for the corresponding theta, phi
-    and t.
-    """
-    c2w = get_translation_t(t)
-    c2w = get_rotation_phi(phi / 180.0 * np.pi) @ c2w
-    c2w = get_rotation_theta(theta / 180.0 * np.pi) @ c2w
-    c2w = np.array([[-1, 0, 0, 0], [0, 0, 1, 0], [0, 1, 0, 0], [0, 0, 0, 1]]) @ c2w
-    return c2w
-
-
 def show_rendered_image(r,theta,phi):
     # Get the camera to world matrix.
     c2w = pose_spherical(theta, phi, r)