Update vis_animals.sh to set model_type for FMPose3D and adjust saved_model_path for consistency with the new model registry structure.

Author: xiu-cs

animals/demo/vis_animals.sh

@@ -7,8 +7,9 @@ sh_file='vis_animals.sh'
 # n_joints=26
 # out_joints=26
 
-model_path='../pre_trained_models/animal3d_pretrained_weights/model_animal3d.py'
-saved_model_path='../pre_trained_models/animal3d_pretrained_weights/CFM_154_4403_best.pth'
+model_type='fmpose3d_animals'
+# model_path=''  # set to a local file path to override the registry
+saved_model_path='../pre_trained_models/fmpose3d_animals/fmpose3d_animals_pretrained_weights.pth'
 
 # path='./images/image_00068.jpg'  # single image
 input_images_folder='./images/'  # folder containing multiple images
@@ -17,7 +18,8 @@ python3 vis_animals.py \
  --type 'image' \
  --path ${input_images_folder} \
  --saved_model_path "${saved_model_path}" \
- --model_path "${model_path}" \
+ ${model_path:+--model_path "$model_path"} \
+ --model_type "${model_type}" \
  --sample_steps ${sample_steps} \
  --batch_size ${batch_size} \
  --layers ${layers} \