Returns the camera name in a scene, eg, cam_00, cam_01, etc.
(
volume_id: T.Optional[int],
scene_id: T.Optional[int],
scene_dir: str = None,
dataset_root_dir: str = 'datasets/hypersim',
)
| 69 | |
| 70 | |
| 71 | def get_hypersim_camera_names( |
| 72 | volume_id: T.Optional[int], |
| 73 | scene_id: T.Optional[int], |
| 74 | scene_dir: str = None, |
| 75 | dataset_root_dir: str = 'datasets/hypersim', |
| 76 | ) -> T.List[str]: |
| 77 | """Returns the camera name in a scene, eg, cam_00, cam_01, etc. """ |
| 78 | if scene_dir is None: |
| 79 | scene_dir, scene_name = get_hypersim_scene_info( |
| 80 | volume_id=volume_id, |
| 81 | scene_id=scene_id, |
| 82 | dataset_root_dir=dataset_root_dir, |
| 83 | ) |
| 84 | # print(f'{os.path.join(scene_dir, "_detail", "cam_*")}') |
| 85 | cam_names = glob(os.path.join(scene_dir, "_detail", "cam_*")) |
| 86 | cam_names = [os.path.basename(n) for n in cam_names] # "cam_00", "cam_01", ... |
| 87 | cam_names = sorted(cam_names) |
| 88 | return cam_names |
| 89 | |
| 90 | |
| 91 | class HypersimDataset: |
no test coverage detected