SD4MetaHuman/server_settings_backup.yaml at main · KaiHangYang/SD4MetaHuman · GitHub

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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from pathlib import Path
from pyrr import Matrix44
import numpy as np

PROJ_DIR = Path(__file__).parent.parent

# ADA_ORTHO_IMAGE_PATH = PROJ_DIR / "HeadMesh/Characters/Ada/Ada_ortho.png"
AIGC_DEFAULT_IMAGE_PATH = PROJ_DIR / "HeadMesh/Cache/default_aigc_image.png"
MIRROR_INFO_PATH = PROJ_DIR / "HeadMesh/Selection/mesh_mirror_info.npy"

EYEEDGE_MIRROR_INFO_PATH = PROJ_DIR / "HeadMesh/Selection/eyeedge_mirror_info.npy"
CARTILAGE_MIRROR_INFO_PATH = PROJ_DIR / "HeadMesh/Selection/cartilage_mirror_info.npy"
EYEBALL_MIRROR_INFO_PATH = PROJ_DIR / "HeadMesh/Selection/eyeball_mirror_info.npy"
TEETH_MIRROR_INFO_PATH = PROJ_DIR / "HeadMesh/Selection/teeth_mirror_info.npy"
SALIVA_MIRROR_INFO_PATH = PROJ_DIR / "HeadMesh/Selection/saliva_mirror_info.npy"

# Only for load.
MESH_LANDMARK_INDICES_PATH = PROJ_DIR / "HeadMesh/Selection/mesh_mediapipe_landmarks_mayamodified.npy"
# Only for save.
MESH_LANDMARK_INDICES_SAVE_PATH = PROJ_DIR / "HeadMesh/PCA/mesh_mediapipe_landmarks.npy"

MESH_HEAD_INDICES_PATH = PROJ_DIR / "HeadMesh/Selection/mesh_head_indices.npy"
MESH_BODY_INDICES_PATH = PROJ_DIR / "HeadMesh/Selection/mesh_body_indices.npy"
MESH_SEAM_INDICES_PATH = PROJ_DIR / "HeadMesh/Selection/mesh_seam_indices.npy"
MESH_HARMONIC_INDICES_PATH = PROJ_DIR / "HeadMesh/Selection/mesh_harmonic_indices.npy"

# For PCA
PCA_FILE_ROOT = Path(PROJ_DIR / f"HeadMesh/PCA/pca_files")
def PCA_FILE_PATH_FUNC(gender, n_comp, suffix=None):
    return PCA_FILE_ROOT / (f"pca_{gender}_{n_comp}"+ (f"_{suffix}.npy" if suffix is not None else ".npy"))

DEFAULT_TEXTURE_IMAGE_PATH = PROJ_DIR / "HeadMesh/Cache/default_head_albedo.png"

# For characters.
CHARACTER_ROOT_DIR = PROJ_DIR / "HeadMesh/Characters"
CHARACTER_INFO_PATH = PROJ_DIR / "HeadMesh/Cache/character_infos.json"
CHARACTERS_DNA_CACHE_PATH = PROJ_DIR / "HeadMesh/Cache/dna_caches.npy"
CHARACTERS_MESH_CACHE_PATH = PROJ_DIR / "HeadMesh/Cache/mesh_caches.npy"

CHARACTER_METAHUMAN_TEXTURE_FILE_NAME = "head_color_map.tga"
CHARACTER_FU_TEXTURE_FILE_NAME = "albedo.png"

# For AIGC
AIGC_ALBEDO_MODEL_ID = PROJ_DIR / "Scripts/resources/model_zoo/diffusers/2946_PBR facial albedo_gas4_4e-6_seed888"
# AIGC_ALBEDO_MODEL_ID = "F:/Projects/AIGC/diffusers/out/2325"

AIGC_ALBEDO_PROMPT_SUFFIX = "(PBR facial albedo:1.9)"
AIGC_ALBEDO_NEGATIVE_SUFFIX = "easynegative, (low quality, worst quality:1.4), bad anatomy, bad composition, out of frame, duplicate, watermark, signature, text, ugly, morbid, mutated, deformed, blurry, disfigured, spotted, bearded, dirty, freckled, (green things:1.2)"

AIGC_PORTRAIT_MODEL_ID = "dreamlike-art/dreamlike-photoreal-2.0"
AIGC_PORTRAIT_PROMPT_SUFFIX = "facial closeup photography, 4k, (neutral expression: 2.0)"
AIGC_PORTRAIT_NEGATIVE_SUFFIX = "easynegative, (low quality, worst quality:1.4), bad anatomy, bad composition, out of frame, duplicate, watermark, signature, text, ugly, morbid, mutated, deformed, blurry, bad anatomy, bad proportions, cloned face, disfigured, fused fingers, fused limbs, too many fingers, long neck, smile, angry, mouth open"

AIGC_DNA_TEMPLATE_MALE_PATH = PROJ_DIR / "Scripts/resources/dnacalib_data/dna_files/Liujing.dna"
AIGC_DNA_TEMPLATE_FEMALE_PATH = PROJ_DIR / "Scripts/resources/dnacalib_data/dna_files/Zhen.dna"

# For the AI models.
NORMAL_SPADE_MODEL_PATH = PROJ_DIR / "Scripts/resources/model_zoo/image_translation/netG_spade_onlyl1loss_20230530.pth"
NORMAL_SPADE_MODEL_STATICS_PATH = PROJ_DIR / "Scripts/resources/model_zoo/image_translation/normal_statistic.npy"
ALBEDO_SR_MODEL_PATH = PROJ_DIR / "Scripts/resources/model_zoo/super_resolution/netG_albedo_r2048_area_TV1e-6_epoch_4_2580.pth"
NORMAL_SR_MODEL_PATH = PROJ_DIR / "Scripts/resources/model_zoo/super_resolution/netG_normal_r2048_area_TV0_epoch_4_3000.pth"

# For the server settings.
SERVER_PCA_FILE_PATH = PROJ_DIR / "HeadMesh/PCA/pca_files" / "pca_female_20_20230621.npy"
SERVER_DNA_TEMPLATE_PATH = (PROJ_DIR / "Scripts/resources/dnacalib_data/dna_files/Ada.dna").as_posix()
SERVER_PORT = 8000
SERVER_PORTRAIT_NUM_STEPS = 60
SERVER_PORTRAIT_GUIDANCE = 7.5
SERVER_ALBEDO_NUM_STEPS = 60
SERVER_ALBEDO_GUIDANCE = 8.0
SERVER_PCA_REGULARIZATION = 0.2
SERVER_IMAGE_MESH_WIDTH = 40
SERVER_IMAGE_MESH_HEIGHT = 40
SERVER_PROJ_MATRIX = Matrix44.perspective_projection(fovy=6.8, aspect=1.0, near=0.1, far=1800.0, dtype=np.float32)
SERVER_VIEW_MATRIX = Matrix44.look_at(eye=[0, 0, 500], target=[0, 0, 0], up=[0, 1, 0], dtype=np.float32)
SERVER_MESH_LANDMAKR_INDICES = np.load(MESH_LANDMARK_INDICES_PATH)
SERVER_SAVE_DNA = True