Benchmark RTMLib DWPose wholebody profile
This commit is contained in:
@@ -167,6 +167,7 @@ services:
|
|||||||
chmod -R a+rX,u+w "$$out"
|
chmod -R a+rX,u+w "$$out"
|
||||||
|
|
||||||
benchmark-rtmlib-body:
|
benchmark-rtmlib-body:
|
||||||
|
profiles: ["validation"]
|
||||||
image: tail2-pose-server:0.1.0
|
image: tail2-pose-server:0.1.0
|
||||||
container_name: tail2-pose-benchmark-rtmlib-body
|
container_name: tail2-pose-benchmark-rtmlib-body
|
||||||
gpus: all
|
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)'
|
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:
|
validate:
|
||||||
profiles: ["validation"]
|
profiles: ["validation"]
|
||||||
image: tail2-pose-server:0.1.0
|
image: tail2-pose-server:0.1.0
|
||||||
|
|||||||
Reference in New Issue
Block a user