pengsida/NeuralBody

1import pickle2import numpy as np3 4 5def read_pickle(pkl_path):6    with open(pkl_path, 'rb') as f:7        u = pickle._Unpickler(f)8        u.encoding = 'latin1'9        return u.load()10 11 12def get_camera(camera_path):13    camera = read_pickle(camera_path)14    K = np.zeros([3, 3])15    K[0, 0] = camera['camera_f'][0]16    K[1, 1] = camera['camera_f'][1]17    K[:2, 2] = camera['camera_c']18    K[2, 2] = 119    R = np.eye(3)20    T = np.zeros([3])21    D = camera['camera_k']22    camera = {'K': K, 'R': R, 'T': T, 'D': D}23    return camera24