admin/Seg_Data_Server_Net
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Expand Seg task coverage and coverage dashboard

Browse Source
This commit is contained in:
admin
2026-06-30 12:54:25 +08:00
parent dd7b7384ec
commit 7a43303f15
18 changed files with 849 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
backend/app/modules/segmodel/tasks.py
View File

@@ -31,11 +31,39 @@ def build_segmodel_task(job_type: str, params: dict, conda_env: str) -> CommandS
script = SEGMODEL_DIR / params.get("script", "2_predict_params_and_FLOPs_V2.py")
return CommandSpec(conda_python(conda_env, script), SEGMODEL_DIR, "calculate SegModel params/FLOPs/FPS")
if job_type == "segmodel.params_flops":
args = conda_python(conda_env, SEGMODEL_DIR / "Tool_get_params_and_FLOPs.py")
append_flag(args, "-a", required(params, "architecture"))
shape = params.get("shape", [512, 512])
if isinstance(shape, str):
shape = [part for part in shape.replace(",", " ").split() if part]
if shape:
args.append("--shape")
args.extend(str(item) for item in shape)
return CommandSpec(args, SEGMODEL_DIR, "calculate SegModel params and FLOPs for one architecture")
if job_type == "segmodel.benchmark":
args = conda_python(conda_env, SEGMODEL_DIR / "Tool_benchmark_smp.py")
append_flag(args, "-a", required(params, "architecture"))
append_flag(args, "-c", params.get("checkpoint"))
shape = params.get("shape", [512, 512])
if isinstance(shape, str):
shape = [part for part in shape.replace(",", " ").split() if part]
if shape:
args.append("--shape")
args.extend(str(item) for item in shape)
append_flag(args, "--repeat-times", params.get("repeat_times"))
append_flag(args, "--log-interval", params.get("log_interval"))
stdin = None if params.get("checkpoint") else params.get("stdin_text")
return CommandSpec(args, SEGMODEL_DIR, "benchmark SegModel FPS/latency", stdin_text=stdin)
if job_type == "segmodel.raw_mask_check":
return CommandSpec(conda_python(conda_env, SEGMODEL_DIR / "1_predict_raw_masks_check.py"), SEGMODEL_DIR, "check SegModel raw mask completeness")
if job_type == "segmodel.metrics":
return CommandSpec(conda_python(conda_env, SEGMODEL_DIR / "3_predict_matrics_from_log.py"), SEGMODEL_DIR, "parse SegModel training/prediction metrics")
return None
if job_type == "segmodel.copy_best":
return CommandSpec(bash(SEGMODEL_DIR / "Tool_Copy_Best_Model.sh"), SEGMODEL_DIR, "copy best SegModel weights to prediction area")
return None