Returns the scene dir and name (eg, "ai_VVV_NNN", "ai_001_001") . Args: volume_id: scene_id: dataset_root_dir: Returns:
(
volume_id: int,
scene_id: int,
dataset_root_dir: str,
)
| 22 | |
| 23 | |
| 24 | def get_hypersim_scene_info( |
| 25 | volume_id: int, |
| 26 | scene_id: int, |
| 27 | dataset_root_dir: str, |
| 28 | ) -> T.Tuple[str, str]: |
| 29 | """ |
| 30 | Returns the scene dir and name (eg, "ai_VVV_NNN", "ai_001_001") . |
| 31 | Args: |
| 32 | volume_id: |
| 33 | scene_id: |
| 34 | dataset_root_dir: |
| 35 | |
| 36 | Returns: |
| 37 | """ |
| 38 | scene_name = f"ai_{volume_id:03d}_{scene_id:03d}" |
| 39 | scene_dir = os.path.join( |
| 40 | dataset_root_dir, |
| 41 | scene_name, |
| 42 | ) |
| 43 | return scene_dir, scene_name |
| 44 | |
| 45 | |
| 46 | def load_hypersim_camera_info( |
no outgoing calls
no test coverage detected