kai/infra-docker-configs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf(unraid): fall back to official RTMW-m

Browse Source
This commit is contained in:
Codex
2026-08-15 10:11:52 -07:00
parent 44b3bc5d2b
commit f77b815fa7
4 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
servers/unraid/tail2-pose-server/app/tail2/main.py
+1 -1
View File
@@ -115,7 +115,7 @@ def capabilities():
"checkpoint": models.det_checkpoint, "checkpoint_sha256": models.sha256(models.det_checkpoint),
"person_class_id": 0, "score_threshold": float(os.getenv("DETECTION_THRESHOLD", ".35")),
"nms_iou_threshold": float(os.getenv("NMS_IOU_THRESHOLD", ".60"))},
"pose": {"name": "rtmw-l-256x192", "framework": "mmpose", "keypoint_schema": "coco_wholebody_133",
"pose": {"name": os.getenv("POSE_NAME", "rtmw-l-256x192"), "framework": "mmpose", "keypoint_schema": "coco_wholebody_133",
"keypoint_order": "MMPose COCO WholeBody 133 official order", "body_schema": "coco17",
"body17_names": BODY17, "config": models.pose_config, "checkpoint": models.pose_checkpoint,
"checkpoint_sha256": models.sha256(models.pose_checkpoint)},
servers/unraid/tail2-pose-server/app/tail2/offline_benchmark.py
+1 -1
View File
@@ -18,7 +18,7 @@ def percentile(values: list[float], p: float) -> float:
def run_offline_benchmark(models, threshold: float, max_people: int) -> None:
path = "/logs/offline-benchmark-" + os.getenv("DETECTOR_NAME", "unknown") + ".json"
path = "/logs/offline-benchmark-" + os.getenv("BENCHMARK_PROFILE", "unknown") + ".json"
if os.path.exists(path):
return
try:
servers/unraid/tail2-pose-server/app/tail2/soak.py
+1 -1
View File
@@ -38,4 +38,4 @@ report = {"duration_seconds": round(time.time()-started, 3), "messages": len(fra
"max_gpu_memory_used_mb": max((x["gpu"]["memory_used_mb"] for x in health_samples), default=None),
"max_gpu_temperature_c": max((x["gpu"]["temperature_c"] for x in health_samples), default=None),
"frames_or_crops_saved": False}
pathlib.Path(f"/validation/soak-{os.getenv('DETECTOR_NAME', 'unknown')}-10m.json").write_text(json.dumps(report, indent=2))
pathlib.Path(f"/validation/soak-{os.getenv('BENCHMARK_PROFILE', 'unknown')}-10m.json").write_text(json.dumps(report, indent=2))