copilot-swe-agent[bot]andkukulu00
ac7ca0f4d8
Agent-Logs-Url: https://github.com/IBB666/IBB_POSE/sessions/c9c0cbf1-b9de-4070-b7ee-8bda69755661 Co-authored-by: kukulu00 <1170651+kukulu00@users.noreply.github.com>
41 lines
1.4 KiB
Bash
41 lines
1.4 KiB
Bash
#!/bin/bash
|
|
# ============================================
|
|
# Author: T. S. Liang @ Rama Alpaca
|
|
# Emails: tsliang2001@gmail.com | shuangliang@ramaalpaca.com | sliang57@connect.hku.hk
|
|
# Date: Aug. 2025
|
|
# Description: Launch script for evaluating SDPose on pose estimation benchmarks.
|
|
# License: MIT License (see LICENSE for details)
|
|
# ============================================
|
|
|
|
LAUNCH_EVAL() {
|
|
|
|
cd ..
|
|
|
|
# ------------------ Configurable Parameters ------------------
|
|
dataset_name='COCO'
|
|
keypoint_scheme='body' # or 'wholebody'
|
|
dataset_root='/data/coding/datasets'
|
|
ann_file='/data/coding/datasets/HumanArt/annotations/validation_humanart.json'
|
|
checkpoint_path='/data/coding/outputs/COCO_813'
|
|
eval_batch_size=16
|
|
dataloader_num_workers=16
|
|
timestep=999
|
|
|
|
# ------------------ Runtime Environment ------------------
|
|
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 accelerate launch --mixed_precision="bf16"\
|
|
--multi_gpu \
|
|
eval/launch_evaluator.py \
|
|
--dataset_name $dataset_name \
|
|
--keypoint_scheme $keypoint_scheme \
|
|
--dataset_root $dataset_root \
|
|
--checkpoint_path $checkpoint_path \
|
|
--ann_file $ann_file \
|
|
--eval_batch_size $eval_batch_size \
|
|
--dataloader_num_workers $dataloader_num_workers \
|
|
--enable_xformers_memory_efficient_attention \
|
|
--timestep $timestep
|
|
}
|
|
|
|
# ------------- Entry -------------
|
|
LAUNCH_EVAL
|