-
Notifications
You must be signed in to change notification settings - Fork 14
Expand file tree
/
Copy pathFMPose3D_test.sh
More file actions
executable file
·34 lines (31 loc) · 930 Bytes
/
FMPose3D_test.sh
File metadata and controls
executable file
·34 lines (31 loc) · 930 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#inference
layers=5
batch_size=1024
sh_file='scripts/FMPose3D_test.sh'
num_hypothesis_list=1
eval_multi_steps=3
topk=8
gpu_id=0
mode='exp'
exp_temp=0.005
folder_name=test_s${eval_multi_steps}_${mode}_h${num_hypothesis_list}_$(date +%Y%m%d_%H%M%S)
model_type='fmpose3d_humans'
# By default, weights are automatically downloaded from Hugging Face Hub.
# To use local weights instead, uncomment the line below:
# model_weights_path='./pre_trained_models/fmpose3d_h36m/FMpose3D_pretrained_weights.pth'
model_weights_path=''
#Test
python3 scripts/FMPose3D_main.py \
--reload \
--topk ${topk} \
--exp_temp ${exp_temp} \
--folder_name ${folder_name} \
--model_weights_path "${model_weights_path}" \
--model_type "${model_type}" \
--eval_sample_steps ${eval_multi_steps} \
--test_augmentation True \
--batch_size ${batch_size} \
--layers ${layers} \
--gpu ${gpu_id} \
--num_hypothesis_list ${num_hypothesis_list} \
--sh_file ${sh_file}