diff --git a/servers/unraid/tail2-pose-server/compose.yaml b/servers/unraid/tail2-pose-server/compose.yaml index d714d05..b7464ce 100644 --- a/servers/unraid/tail2-pose-server/compose.yaml +++ b/servers/unraid/tail2-pose-server/compose.yaml @@ -167,6 +167,7 @@ services: chmod -R a+rX,u+w "$$out" benchmark-rtmlib-body: + profiles: ["validation"] image: tail2-pose-server:0.1.0 container_name: tail2-pose-benchmark-rtmlib-body gpus: all @@ -198,6 +199,38 @@ services: - | python -c 'from tail2.models import Models; from tail2.offline_benchmark import run_offline_benchmark; m=Models(); m.load(); run_offline_benchmark(m, .35, 4)' + benchmark-rtmlib-wholebody: + image: tail2-pose-server:0.1.0 + container_name: tail2-pose-benchmark-rtmlib-wholebody + gpus: all + restart: "no" + entrypoint: ["/bin/bash", "-lc"] + depends_on: + models: + condition: service_completed_successfully + volumes: + - /mnt/user/appdata/tail2-pose-server/app:/app:ro + - /mnt/user/appdata/tail2-pose-server/models:/models:ro + - ./validation:/validation + environment: + PYTHONPATH: /app + CUDA_VISIBLE_DEVICES: "1" + MODEL_BACKEND: rtmlib_wholebody + BENCHMARK_PROFILE: rtmlib_wholebody_yolox-m_dwpose-m_gpu1_threads2_iobinding + BENCHMARK_OUTPUT: /validation/benchmark-rtmlib-wholebody-iobinding.json + DETECTION_THRESHOLD: "0.35" + NMS_IOU_THRESHOLD: "0.60" + MAX_PEOPLE: "4" + DET_CHECKPOINT: /models/yolox-m-humanart.onnx + POSE_CHECKPOINT: /models/dwpose-m-wholebody.onnx + OMP_NUM_THREADS: "2" + MKL_NUM_THREADS: "2" + OPENBLAS_NUM_THREADS: "2" + CV2_NUM_THREADS: "2" + command: + - | + python -c 'from tail2.models import Models; from tail2.offline_benchmark import run_offline_benchmark; m=Models(); m.load(); run_offline_benchmark(m, .35, 4)' + validate: profiles: ["validation"] image: tail2-pose-server:0.1.0