diff --git a/servers/unraid/tail2-pose-server/compose.yaml b/servers/unraid/tail2-pose-server/compose.yaml index 1579cd6..336a92d 100644 --- a/servers/unraid/tail2-pose-server/compose.yaml +++ b/servers/unraid/tail2-pose-server/compose.yaml @@ -112,10 +112,10 @@ services: SAVE_CROPS: "false" LOG_COORDINATES: "false" RUN_STARTUP_BENCHMARK: "false" - OMP_NUM_THREADS: "2" - MKL_NUM_THREADS: "2" - OPENBLAS_NUM_THREADS: "2" - CV2_NUM_THREADS: "2" + OMP_NUM_THREADS: "4" + MKL_NUM_THREADS: "4" + OPENBLAS_NUM_THREADS: "4" + CV2_NUM_THREADS: "4" CUDA_VISIBLE_DEVICES: "1" DETECTOR_NAME: yolox-m-humanart-coco-640x640 DET_CONFIG: "" @@ -233,6 +233,7 @@ 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-threads: + profiles: ["validation"] image: tail2-pose-server:0.1.0 container_name: tail2-pose-benchmark-threads gpus: all @@ -263,6 +264,40 @@ 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)' done + benchmark-openmmlab: + image: tail2-pose-server:0.1.0 + container_name: tail2-pose-benchmark-openmmlab + 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:/models/mmdetection:/models/mmpose + CUDA_VISIBLE_DEVICES: "1" + MODEL_BACKEND: openmmlab + BENCHMARK_PROFILE: openmmlab_rtmdet-tiny_rtmw-m_gpu1_threads4 + BENCHMARK_OUTPUT: /validation/benchmark-openmmlab.json + DETECTION_THRESHOLD: "0.35" + NMS_IOU_THRESHOLD: "0.60" + MAX_PEOPLE: "4" + DET_CONFIG: /models/mmdetection/configs/rtmdet/rtmdet_tiny_8xb32-300e_coco.py + DET_CHECKPOINT: /models/rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth + POSE_CONFIG: /models/mmpose/configs/wholebody_2d_keypoint/rtmpose/cocktail14/rtmw-m_8xb1024-270e_cocktail14-256x192.py + POSE_CHECKPOINT: /models/rtmw-dw-l-m_simcc-cocktail14_270e-256x192-20231122.pth + OMP_NUM_THREADS: "4" + MKL_NUM_THREADS: "4" + OPENBLAS_NUM_THREADS: "4" + CV2_NUM_THREADS: "4" + 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