Benchmark OpenMMLab baseline with 300 samples

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