LinzhouLi commited on
Commit
4747cf0
·
verified ·
1 Parent(s): 8bbe9c7

Simplify model card to project and checkpoint basics

Browse files
Files changed (2) hide show
  1. README.md +12 -77
  2. checksums.sha256 +1 -1
README.md CHANGED
@@ -9,94 +9,29 @@ tags:
9
  - flame
10
  - dinov3
11
  - arxiv:2608.09238
12
- datasets:
13
- - LinzhouLi/RealDenseFace-Ava256
14
  ---
15
 
16
  # RealDenseFace
17
 
18
  **Real-time Monocular 3D Face Reconstruction from Dense UV-space Priors**
19
 
20
- Linzhou Li, Tianjia Shao, Kun Zhou · State Key Lab of CAD&CG, Zhejiang University
 
21
 
22
- [Project](https://gapszju.github.io/RealDenseFace/) · [Paper](https://arxiv.org/abs/2608.09238) · [Code](https://github.com/LinzhouLi/RealDenseFace) · [Ava-256 training data](https://huggingface.co/datasets/LinzhouLi/RealDenseFace-Ava256)
23
 
24
- ## Overview
25
 
26
- RealDenseFace reconstructs a FLAME face from a monocular RGB image in two stages. A DINOv3-based network predicts dense UV-to-image correspondences, relative depth, and their log-variances. Predictions sampled at FLAME vertex UV coordinates provide targets and confidence weights for a tailored CUDA Gauss-Newton solver.
27
 
28
- The code supports single-image fitting, offline monocular sequence reconstruction, online tracking, and calibrated multi-view fitting. This repository hosts the pretrained dense-prior networks and their matching configurations; the inference and fitting implementation is maintained on GitHub.
 
 
 
29
 
30
- ## Checkpoints
31
 
32
- | Variant | Encoder | Input resolution | Checkpoint | Typical use |
33
- | --- | --- | --- | --- | --- |
34
- | ViT-S | DINOv3 ViT-S+ | 512 x 512 | `weights/realdenseface/vits.pth` | Default online tracking model |
35
- | ViT-B | DINOv3 ViT-B | 512 x 512 | `weights/realdenseface/vitb.pth` | Default single-image and offline model |
36
-
37
- The `.pth` files are PyTorch state dictionaries consumed by the official code. They are not Transformers checkpoints. Both model configurations use `scale_coord: 1.0` and `scale_depth: 8.0`; the official inferencer handles normalization and output rescaling.
38
-
39
- ## Download and run
40
-
41
- Clone the implementation and run these commands from its root:
42
-
43
- ```bash
44
- git clone https://github.com/LinzhouLi/RealDenseFace.git
45
- cd RealDenseFace
46
- pip install -U huggingface_hub
47
- hf download LinzhouLi/RealDenseFace --local-dir .
48
- ```
49
-
50
- For a private repository, authenticate first with `hf auth login` using an account with access.
51
-
52
- Set up Python 3.11, a CUDA-enabled PyTorch build, and a compatible NVIDIA GPU/CUDA toolkit. Install the remaining dependencies and compile the solver:
53
-
54
- ```bash
55
- pip install -r requirements.txt
56
- pip install -e ./cuda_extensions/flame_solver --no-build-isolation
57
- ```
58
-
59
- Before running, obtain the additional resources described in the [GitHub installation instructions](https://github.com/LinzhouLi/RealDenseFace#installation):
60
-
61
- ```text
62
- weights/facebox/face_box.pth
63
- weights/flame/flame_assets.npz
64
- weights/flame/flame2023.pkl
65
- ```
66
-
67
- These resources are not included in this model repository. Obtain the standard FLAME 2023 model separately from the [official FLAME website](https://flame.is.tue.mpg.de/), subject to its license.
68
-
69
- Single-image fitting with ViT-B:
70
-
71
- ```bash
72
- python fit_single_image.py --input path/to/image.jpg --output_npz output/face.npz --output_image output/face.jpg
73
- ```
74
-
75
- To use ViT-S, add `--model_config configs/model/vits.yaml --model_weights weights/realdenseface/vits.pth`.
76
-
77
- Online video tracking with ViT-S:
78
-
79
- ```bash
80
- python track_video_online.py --input assets/demo_video.mp4 --output_mp4 output/tracking.mp4
81
- ```
82
-
83
- Outputs include fitted FLAME parameters and optional reconstruction visualizations. See the GitHub README for offline sequence and multi-view examples.
84
-
85
- ## Training data and evaluation
86
-
87
- The paper describes training on processed AVA-256, NPHM, and FaceScape data. The [RealDenseFace-Ava256 dataset](https://huggingface.co/datasets/LinzhouLi/RealDenseFace-Ava256) provides the processed Ava-256 images and registrations; it is one component of the training data.
88
-
89
- The paper reports approximately 0.18 seconds for single-image fitting with unknown intrinsics, 84 FPS online tracking, and approximately 24x faster offline sequence processing than VHAP on an RTX 4090. These are paper-reported results, not measurements performed during this upload. Runtime depends on hardware, compilation, model variant, and fitting settings.
90
-
91
- ## Usage notes
92
-
93
- The output geometry is constrained by the FLAME model. Camera calibration, image cropping, and the quality of the predicted dense priors affect reconstruction. The official task configurations differ: the single-image and offline configurations use relative-depth weight 2.0, while the online and NeRSemble multi-view configurations use weight 0.0. Use the matching configuration when reproducing a task.
94
-
95
- The model and fitting code require the custom runtime described above; uploading the checkpoints does not by itself provide a hosted inference API or interactive demo.
96
-
97
- ## License
98
-
99
- This repository uses the MIT license selected for this release; see [LICENSE](LICENSE). Third-party components, including DINOv3, FaceBox, FLAME, and the source datasets, retain their respective licenses. This repository does not redistribute the FLAME model or grant rights to those external resources. The released Ava-256 dataset has its own CC BY-NC 4.0 terms.
100
 
101
  ## Citation
102
 
@@ -107,4 +42,4 @@ This repository uses the MIT license selected for this release; see [LICENSE](LI
107
  journal = {arXiv preprint arXiv:2608.09238},
108
  year = {2026}
109
  }
110
- ```
 
9
  - flame
10
  - dinov3
11
  - arxiv:2608.09238
 
 
12
  ---
13
 
14
  # RealDenseFace
15
 
16
  **Real-time Monocular 3D Face Reconstruction from Dense UV-space Priors**
17
 
18
+ Linzhou Li, Tianjia Shao, Kun Zhou
19
+ State Key Lab of CAD&CG, Zhejiang University
20
 
21
+ [Project](https://gapszju.github.io/RealDenseFace/) · [Paper](https://arxiv.org/abs/2608.09238) · [Code](https://github.com/LinzhouLi/RealDenseFace)
22
 
23
+ RealDenseFace combines dense UV-space correspondence and relative-depth predictions with a tailored CUDA Gauss-Newton solver for FLAME reconstruction. It supports single-image fitting, offline sequence reconstruction, online tracking, and calibrated multi-view fitting.
24
 
25
+ ## Models
26
 
27
+ | Model | Encoder | Input size | Checkpoint |
28
+ | --- | --- | --- | --- |
29
+ | ViT-S | DINOv3 ViT-S+ | 512 × 512 | [vits.pth](weights/realdenseface/vits.pth) |
30
+ | ViT-B | DINOv3 ViT-B | 512 × 512 | [vitb.pth](weights/realdenseface/vitb.pth) |
31
 
32
+ ViT-S is the default model for online tracking; ViT-B is the default for single-image and offline reconstruction. Matching model and fitting configurations are included in `configs/`.
33
 
34
+ For installation and usage, see the [GitHub repository](https://github.com/LinzhouLi/RealDenseFace). FLAME model files must be obtained separately from the [official website](https://flame.is.tue.mpg.de/) under its license.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
35
 
36
  ## Citation
37
 
 
42
  journal = {arXiv preprint arXiv:2608.09238},
43
  year = {2026}
44
  }
45
+ ```
checksums.sha256 CHANGED
@@ -5,6 +5,6 @@ e0433a66441abcad8229fb93c373b81707a24659ddf7a7d9c8d6e7226dbb98e4 configs/fittin
5
  8c1cced899f4409c71f3e50e174d312a161fd26982aad0894ddde4a30246a6ab configs/model/vitb.yaml
6
  1f02b837a62dfa1959b6e09be23332441eacf6a74892d50c0e18d569eeec8686 configs/model/vits.yaml
7
  718974a99973b53f9cfce42dc1528368b6cbf6b7d3cc97981269c9696fb5b966 LICENSE
8
- 502f3280e21d32bd80304799ce8b7da80936520305389037d8215c63fd16d6d1 README.md
9
  3b3bbd026c277aae561aed0047f43bbae2833c000779904f6dd73e97e321c604 weights/realdenseface/vitb.pth
10
  9644fb9c7b7b8b5b6ca0d06fdebae418c154f1f4224a1ef5298777cc399ea15c weights/realdenseface/vits.pth
 
5
  8c1cced899f4409c71f3e50e174d312a161fd26982aad0894ddde4a30246a6ab configs/model/vitb.yaml
6
  1f02b837a62dfa1959b6e09be23332441eacf6a74892d50c0e18d569eeec8686 configs/model/vits.yaml
7
  718974a99973b53f9cfce42dc1528368b6cbf6b7d3cc97981269c9696fb5b966 LICENSE
8
+ 7ed3d97070800600c8fa59a8c31eb277bf819375fca538300099ed9b41175917 README.md
9
  3b3bbd026c277aae561aed0047f43bbae2833c000779904f6dd73e97e321c604 weights/realdenseface/vitb.pth
10
  9644fb9c7b7b8b5b6ca0d06fdebae418c154f1f4224a1ef5298777cc399ea15c weights/realdenseface/vits.pth