Enable dedicated MMSeg full mmcv runtime

This commit is contained in:
2026-06-30 13:23:32 +08:00
parent 4d80ec4d75
commit 2d7d54ba13
10 changed files with 121 additions and 35 deletions

View File

@@ -20,6 +20,7 @@ class Settings:
log_dir: Path
weights_root: Path
task_conda_env: str
mmseg_conda_env: str
backend_conda_env: str
weight_mode: str
enable_shell_tasks: bool
@@ -46,6 +47,7 @@ def get_settings() -> Settings:
log_dir=log_dir,
weights_root=weights_root,
task_conda_env=os.getenv("SEG_TASK_CONDA_ENV", "seg_smp"),
mmseg_conda_env=os.getenv("SEG_MMSEG_CONDA_ENV", "seg_mmcv"),
backend_conda_env=os.getenv("SEG_BACKEND_CONDA_ENV", "seg_smp"),
weight_mode=os.getenv("SEG_WEIGHT_MODE", "copy"),
enable_shell_tasks=os.getenv("SEG_ENABLE_SHELL_TASKS", "1") == "1",