Hugging Face's logo

Pampaj
/
wbes-faceembedding-dn-mixed-topology-v1

PyTorch
3d
mesh
face-recognition
diffusionnet
topology-robustness
Model card Files
xet
 Community
wbes-faceembedding-dn-mixed-topology-v1
File size: 3,787 Bytes
c8e42d7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
{
  "args": {
    "batch_subjects": 5,
    "data_dir": "/equilibrium/lpampaloni/WBES-FaceEmbedding/datasets/REMESH/npz_data_topo_500_withops",
    "device": "cuda",
    "dist_npz": "/equilibrium/lpampaloni/WBES-FaceEmbedding/face_embedding/gt_encdec/autoencoder/latent_analysis/gt_distance_matrix/normalized_matrix_distances.npz",
    "dropout": 0.1,
    "eig_k": 300,
    "epochs": 120,
    "eps": 1e-08,
    "eval_every": 2,
    "eval_mode": "average",
    "grad_clip": 1.0,
    "init_checkpoint": "",
    "k_spec": 0,
    "lambda_id": 0.25,
    "lambda_mesh": 1.0,
    "lambda_rank": 0.5,
    "lambda_smooth": 0.01,
    "lambda_subject": 1.0,
    "lambda_teacher": 0.0,
    "latent_dim": 256,
    "log_spec": false,
    "lr": 0.0001,
    "max_meshes_per_subject_eval": 6,
    "max_meshes_per_subject_train": 6,
    "max_subjects_eval_train": 16,
    "model": "xyz_dn",
    "n_blocks": 4,
    "n_hks": 0,
    "n_sigma_eval": 6,
    "n_wks": 0,
    "noise_mode_weights": "translation=4,rotation=2,jitter=1",
    "noise_modes": "translation,rotation,jitter",
    "outlier_frac": 0.02,
    "outlier_scale": 6.0,
    "p_noise": 0.6,
    "pool_mode": "meanmax",
    "preload_eval_samples_train": true,
    "preload_eval_workers_train": 2,
    "rank_hard_frac": 0.7,
    "rank_margin": 0.05,
    "rank_pairs": 1024,
    "rank_tau": 0.02,
    "rigid_rot_deg": 12.0,
    "rigid_rot_deg_min": 0.5,
    "rigid_trans_scale": 0.03,
    "rigid_trans_scale_min": 0.001,
    "runs_root": "/equilibrium/lpampaloni/WBES-FaceEmbedding/face_embedding/gt_encdec/remeshing/intrinsic/newdata/dn_mixed_topology_v1",
    "save_every": 5,
    "seed": 1234,
    "sigma_max": 0.02,
    "sigma_max_eval": 0.1,
    "sigma_min": 0.0005,
    "sigma_min_eval": 0.001,
    "teacher_checkpoint": "",
    "train_latent_noise": true,
    "train_level": "mixed",
    "train_pair_mode": "cross_topology",
    "use_id_loss": true,
    "use_smooth_loss": false,
    "use_spectrum": true,
    "use_xyz": true,
    "weight_decay": 1e-06,
    "width": 128,
    "xyz_feature_dropout": 0.0
  },
  "fingerprint": {
    "batch_subjects": 5,
    "dropout": 0.1,
    "eig_k": 300,
    "epochs": 120,
    "eval_mode": "average",
    "init_checkpoint": "",
    "k_spec": 0,
    "lambda_id": 0.25,
    "lambda_mesh": 1.0,
    "lambda_rank": 0.5,
    "lambda_smooth": 0.01,
    "lambda_subject": 1.0,
    "lambda_teacher": 0.0,
    "latent_dim": 256,
    "log_spec": false,
    "lr": 0.0001,
    "max_meshes_per_subject_eval": 6,
    "max_meshes_per_subject_train": 6,
    "max_subjects_eval_train": 16,
    "model": "xyz_dn",
    "n_blocks": 4,
    "n_hks": 0,
    "n_sigma_eval": 6,
    "n_wks": 0,
    "noise_mode_probs": {
      "jitter": 0.14285714285714285,
      "rotation": 0.2857142857142857,
      "translation": 0.5714285714285714
    },
    "noise_modes": [
      "translation",
      "rotation",
      "jitter"
    ],
    "outlier_frac": 0.02,
    "outlier_scale": 6.0,
    "p_noise": 0.6,
    "pool_mode": "meanmax",
    "preload_eval_samples_train": true,
    "preload_eval_workers_train": 2,
    "rank_hard_frac": 0.7,
    "rank_margin": 0.05,
    "rank_pairs": 1024,
    "rank_tau": 0.02,
    "rigid_rot_deg": 12.0,
    "rigid_rot_deg_min": 0.5,
    "rigid_trans_scale": 0.03,
    "rigid_trans_scale_min": 0.001,
    "seed": 1234,
    "sigma_max": 0.02,
    "sigma_max_eval": 0.1,
    "sigma_min": 0.0005,
    "sigma_min_eval": 0.001,
    "teacher_checkpoint": "",
    "train_latent_noise": true,
    "train_level": "mixed",
    "train_pair_mode": "cross_topology",
    "use_id_loss": true,
    "use_smooth_loss": false,
    "use_spectrum": true,
    "use_xyz": true,
    "weight_decay": 1e-06,
    "width": 128,
    "xyz_feature_dropout": 0.0
  },
  "noise_modes": [
    "translation",
    "rotation",
    "jitter"
  ]
}