Function render_path_spiral

dataset_loaders/load_7Scenes.py:199–208

(c2w, up, rads, focal, zdelta, zrate, rots, N)

    return poses_centered, pose_avg #np.linalg.inv(pose_avg_homo)

def render_path_spiral(c2w, up, rads, focal, zdelta, zrate, rots, N):
    render_poses = []
    rads = np.array(list(rads) + [1.])
    hwf = c2w[:,4:5] # it's empty here...
    
    for theta in np.linspace(0., 2. * np.pi * rots, N+1)[:-1]:
        c = np.dot(c2w[:3,:4], np.array([np.cos(theta), -np.sin(theta), -np.sin(theta*zrate), 1.]) * rads) 
        z = normalize(c - np.dot(c2w[:3,:4], np.array([0,0,-focal, 1.])))
        render_poses.append(np.concatenate([viewmatrix(z, up, c), hwf], 1))
    return render_poses

def generate_render_pose(poses, bds):
    idx = np.random.choice(poses.shape[0])