Spaces:
Build error
Build error
File size: 1,806 Bytes
631c942 cf92dec 970f13d cf92dec 20b0829 cf92dec 970f13d b718883 970f13d 60e5037 cf92dec 522bbb9 cf92dec | 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 35 36 37 38 39 40 41 42 43 | import os
import json
from pathlib import Path
from environs import Env
from huggingface_hub import hf_hub_download, snapshot_download
env = Env(expand_vars=True)
env_file_path = Path(f"{Path.home()}/.config/pixel3dmm/.env")
if env_file_path.exists():
env.read_env(str(env_file_path), recurse=False)
with env.prefixed("PIXEL3DMM_"):
CODE_BASE = env("CODE_BASE")
PREPROCESSED_DATA = env("PREPROCESSED_DATA")
TRACKING_OUTPUT = env("TRACKING_OUTPUT")
base = snapshot_download(
repo_id="alexnasa/pixel3dmm", # your model repo
repo_type="model", # model vs dataset
)
FLAME_ASSET = os.path.join(base, "generic_model.pkl")
MICA_TAR_ASSET = os.path.join(base, "mica.tar")
PIPNET_LOCAL_ASSET= os.path.join(base, "epoch59.pth")
CKPT_N_PRED = os.path.join(base, "normals.ckpt")
CKPT_UV_PRED = os.path.join(base, "uv.ckpt")
ANT_DIR = os.path.join(base, "insightface")
BUFFALO_DIR = os.path.join(base, "insightface")
head_template = f'{CODE_BASE}/assets/head_template.obj'
head_template_color = f'{CODE_BASE}/assets/head_template_color.obj'
head_template_ply = f'{CODE_BASE}/assets/test_rigid.ply'
VALID_VERTICES_WIDE_REGION = f'{CODE_BASE}/assets/uv_valid_verty_noEyes_debug.npy'
VALID_VERTS_UV_MESH = f'{CODE_BASE}/assets/uv_valid_verty.npy'
VERTEX_WEIGHT_MASK = f'{CODE_BASE}/assets/flame_vertex_weights.npy'
MIRROR_INDEX = f'{CODE_BASE}/assets/flame_mirror_index.npy'
EYE_MASK = f'{CODE_BASE}/assets/uv_mask_eyes.png'
FLAME_UV_COORDS = f'{CODE_BASE}/assets/flame_uv_coords.npy'
VALID_VERTS_NARROW = f'{CODE_BASE}/assets/uv_valid_verty_noEyes.npy'
VALID_VERTS = f'{CODE_BASE}/assets/uv_valid_verty_noEyes_noEyeRegion_debug_wEars.npy'
FLAME_MASK_ASSET = f'{CODE_BASE}/src/pixel3dmm/preprocessing/MICA/data/'
|