cvcg_utils.misc package
Submodules
cvcg_utils.misc.image module
cvcg_utils.misc.image_proc module
cvcg_utils.misc.pickle_io module
- cvcg_utils.misc.pickle_io.dump_pickle(path, data)
- cvcg_utils.misc.pickle_io.load_pickle(path)
cvcg_utils.misc.slurm_utils module
- cvcg_utils.misc.slurm_utils.gen_slurm_scripts(script_name: str, job_name: str, time: str, partition: str, n_nodes: int, n_cpus: int, mem: str, n_gpus: int, commands: str, exclude_nodes: str = None)
cvcg_utils.misc.tensor_utils module
- cvcg_utils.misc.tensor_utils.np2cpu(array: ndarray) torch.Tensor
- cvcg_utils.misc.tensor_utils.np2cuda(array: ndarray) torch.Tensor
- cvcg_utils.misc.tensor_utils.th2np(tensor: torch.Tensor) ndarray
cvcg_utils.misc.video module
cvcg_utils.misc.vis_utils module
- cvcg_utils.misc.vis_utils.gen_camera_set_visualization(K_list: List[ndarray], c2w_list: List[ndarray], H: int | List[int], W: int | List[int], cone_height: float)
all in opencv format K: 3x3 c2w: 4x4
- cvcg_utils.misc.vis_utils.gen_camera_visualization(K: ndarray, c2w: ndarray, H: int, W: int, cone_height: float)
all in opencv format K: 3x3 c2w: 4x4
- cvcg_utils.misc.vis_utils.gen_line_visualization(start: ndarray, end: ndarray)
3d vectors
- cvcg_utils.misc.vis_utils.gen_quad_cone(K: ndarray, c2w: ndarray, quad_corners: ndarray, cone_height: float)
very similar to gen_camera_visualization, but only a patch of it specified by quad_corners
all in opencv format K: 3x3 c2w: 4x4 quad_corners: [4, 2], top-left, top right, bottom right, bottom left, in pixel coordinates
- cvcg_utils.misc.vis_utils.get_voxel_visualization(voxel_centers, side_length, voxel_colors=None)