(
self,
project_root: str | Path,
image_root: str | Path,
model_config_path: str | Path,
)
| 51 | """ |
| 52 | |
| 53 | def __init__( |
| 54 | self, |
| 55 | project_root: str | Path, |
| 56 | image_root: str | Path, |
| 57 | model_config_path: str | Path, |
| 58 | ) -> None: |
| 59 | self.project_root = Path(project_root) |
| 60 | self.image_root = Path(image_root) |
| 61 | self.model_config_path = Path(model_config_path) |
| 62 | self.model_cfg = config_utils.read_config_as_dict(str(model_config_path)) |
| 63 | self.pose_task = Task(self.model_cfg["method"]) |
| 64 | self._loaded_data: dict[str, dict[str, list[dict]]] = {} |
| 65 | |
| 66 | @property |
| 67 | def model_folder(self) -> Path: |