Spaces:
Running on Zero
Running on Zero
| """Utility helpers for the accident detection module.""" | |
| from __future__ import annotations | |
| import os | |
| import random | |
| import sys | |
| from pathlib import Path | |
| from typing import Any | |
| import numpy as np | |
| import torch | |
| # Reproducible paths matching the project layout. | |
| PROJECT_ROOT = Path(__file__).resolve().parents[2] | |
| DATASET_DIR = PROJECT_ROOT / "dataset" | |
| PROCESSED_DIR = PROJECT_ROOT / "data" / "processed" / "accident" | |
| MODEL_DIR = PROJECT_ROOT / "models" / "accident" | |
| REPORT_DIR = PROJECT_ROOT / "reports" / "accident" | |
| FIGURE_DIR = PROJECT_ROOT / "reports" / "figures" / "accident" | |
| RANDOM_SEED = 42 | |
| def seed_everything(seed: int = RANDOM_SEED) -> None: | |
| """Set deterministic seeds for Python, NumPy, and PyTorch.""" | |
| random.seed(seed) | |
| np.random.seed(seed) | |
| torch.manual_seed(seed) | |
| os.environ["PYTHONHASHSEED"] = str(seed) | |
| if torch.cuda.is_available(): | |
| torch.cuda.manual_seed_all(seed) | |
| def get_device() -> torch.device: | |
| """Return the best available torch device (CPU only in this phase).""" | |
| return torch.device("cpu") | |
| def get_env_info() -> dict[str, Any]: | |
| """Return a dictionary of environment and package versions.""" | |
| import platform | |
| import PIL | |
| import pyarrow | |
| import torchvision | |
| return { | |
| "python_version": sys.version, | |
| "platform": platform.platform(), | |
| "processor": platform.processor(), | |
| "torch_version": torch.__version__, | |
| "torchvision_version": torchvision.__version__, | |
| "cuda_available": torch.cuda.is_available(), | |
| "pillow_version": PIL.__version__, | |
| "pyarrow_version": pyarrow.__version__, | |
| "numpy_version": np.__version__, | |
| } | |