Simplify model card to project and checkpoint basics
Browse files- README.md +12 -77
- 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
|
|
|
|
| 21 |
|
| 22 |
-
[Project](https://gapszju.github.io/RealDenseFace/) · [Paper](https://arxiv.org/abs/2608.09238) · [Code](https://github.com/LinzhouLi/RealDenseFace)
|
| 23 |
|
| 24 |
-
|
| 25 |
|
| 26 |
-
|
| 27 |
|
| 28 |
-
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
-
|
| 31 |
|
| 32 |
-
|
| 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 |
-
|
| 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
|