alexnasa commited on
Commit
7581429
·
verified ·
1 Parent(s): 0623db6

Update src/pixel3dmm/env_paths.py

Browse files
Files changed (1) hide show
  1. src/pixel3dmm/env_paths.py +2 -2
src/pixel3dmm/env_paths.py CHANGED
@@ -14,7 +14,7 @@ with env.prefixed("PIXEL3DMM_"):
14
  PREPROCESSED_DATA = env("PREPROCESSED_DATA")
15
  TRACKING_OUTPUT = env("TRACKING_OUTPUT")
16
 
17
- flame_local_path = hf_hub_download(
18
  repo_id="alexnasa/pixel3dmm",
19
  filename="generic_model.pkl",
20
  repo_type="model",
@@ -49,7 +49,7 @@ EYE_MASK = f'{CODE_BASE}/assets/uv_mask_eyes.png'
49
  FLAME_UV_COORDS = f'{CODE_BASE}/assets/flame_uv_coords.npy'
50
  VALID_VERTS_NARROW = f'{CODE_BASE}/assets/uv_valid_verty_noEyes.npy'
51
  VALID_VERTS = f'{CODE_BASE}/assets/uv_valid_verty_noEyes_noEyeRegion_debug_wEars.npy'
52
- FLAME_ASSETS = f'{CODE_BASE}/src/pixel3dmm/preprocessing/MICA/data/'
53
 
54
  # paths to pretrained pixel3dmm checkpoints
55
  CKPT_UV_PRED = f'{CODE_BASE}/pretrained_weights/uv.ckpt'
 
14
  PREPROCESSED_DATA = env("PREPROCESSED_DATA")
15
  TRACKING_OUTPUT = env("TRACKING_OUTPUT")
16
 
17
+ FLAME_ASSETS = hf_hub_download(
18
  repo_id="alexnasa/pixel3dmm",
19
  filename="generic_model.pkl",
20
  repo_type="model",
 
49
  FLAME_UV_COORDS = f'{CODE_BASE}/assets/flame_uv_coords.npy'
50
  VALID_VERTS_NARROW = f'{CODE_BASE}/assets/uv_valid_verty_noEyes.npy'
51
  VALID_VERTS = f'{CODE_BASE}/assets/uv_valid_verty_noEyes_noEyeRegion_debug_wEars.npy'
52
+
53
 
54
  # paths to pretrained pixel3dmm checkpoints
55
  CKPT_UV_PRED = f'{CODE_BASE}/pretrained_weights/uv.ckpt'