update: bump model version

lib/modeling/pipelines/hsmr.py

@@ -1,7 +1,5 @@
 from lib.kits.basic import *
 
-import traceback
-
 from lib.utils.vis import Wis3D
 from lib.utils.vis.py_renderer import render_mesh_overlay_img
 from lib.utils.data import to_tensor
@@ -35,7 +33,7 @@ def build_inference_pipeline(
     pipeline.set_data_adaption(data_module_name='IMG_PATCHES')
     # 2.2. Load the checkpoint.
     if ckpt_fn is None:
-        ckpt_fn = model_root / 'checkpoints' / 'last.ckpt'
+        ckpt_fn = model_root / 'checkpoints' / 'hsmr.ckpt'
     pipeline.load_state_dict(torch.load(ckpt_fn, map_location=device)['state_dict'])
     get_logger(brief=True).info(f'Load checkpoint from {ckpt_fn}.')
 

lib/version.py

@@ -1,3 +1,3 @@
 # coding=utf-8
 __version__ = 'v1.0'
-DEFAULT_HSMR_ROOT = 'data_inputs/released_models/HSMR-ViTH-r1d0'
+DEFAULT_HSMR_ROOT = 'data_inputs/released_models/HSMR-ViTH-r1d1'

tools/prepare_data.sh

@@ -6,5 +6,5 @@ tar -xzf body_models.tar.gz
 mkdir -p released_models
 cd released_models
 echo "[SCRIPT-LOG] Downloading checkpoints..." ; date
-wget -q -c 'https://huggingface.co/IsshikiHugh/HSMR-data_inputs/resolve/main/released_models/HSMR-ViTH-r1d0.tar.gz' -O HSMR-ViTH-r1d0.tar.gz
-tar -xzf HSMR-ViTH-r1d0.tar.gz
+wget -q -c 'https://huggingface.co/IsshikiHugh/HSMR-data_inputs/resolve/main/released_models/HSMR-ViTH-r1d1.tar.gz' -O HSMR-ViTH-r1d1.tar.gz
+tar -xzf HSMR-ViTH-r1d1.tar.gz