INNER CODE UNIT · Python

colorbar

yashbhalgat/HashNeRF-pytorch · camera_pose_visualizer.py:47

    def colorbar(self, max_frame_length):
        cmap = mpl.cm.rainbow
        norm = mpl.colors.Normalize(vmin=0, vmax=max_frame_length)
        self.fig.colorbar(mpl.cm.ScalarMappable(norm=norm, cmap=cmap), orientation='vertical', label='Frame Number')

    def show(self):
        plt.title('Extrinsic Parameters')
        plt.show()

if __name__ == '__main__':
    poses = []
    with open(os.path.join('data/nerf_synthetic/chair/', 'transforms_train.json'), 'r') as fp:
        meta = json.load(fp)
        for frame in meta['frames']:
            poses.append(np.array(frame['transform_matrix']))
    t_arr = np.array([pose[:3,-1] for pose in poses])
    maxes = t_arr.max(axis=0)
    mins = t_arr.min(axis=0)

View source record →