-
Update README.md
-
Update README.md
-
Update README.md
# ViTPain: Pretrained Vision Transformer for Pain Assessment
Pretrained checkpoint for **ViTPain**, a reference-guided Vision Transformer for automated pain intensity assessment. Trained on the [3D-Pain synthetic dataset](https://huggingface.co/datasets/SoroushMehraban/3D-Pain). Use this checkpoint to fine-tune on real pain datasets (e.g. UNBC-McMaster).
## Model Details
- **Architecture**: DinoV3-large backbone + LoRA (rank=8, alpha=16)
- **Task**: PSPI regression (0–16) and Action Unit prediction
- **Training**: 3D-Pain synthetic faces, 150 epochs
- **Best checkpoint**: epoch 141, validation MAE 1.859
- **Input**: 224×224 RGB face image + optional neutral reference image
## Download
```bash
pip install huggingface-hub
huggingface-cli download xinlei55555/ViTPain vitpain-epoch=141-val_regression_mae=1.859.ckpt --local-dir ./
```
Or in Python:
```python
from huggingface_hub import hf_hub_download
checkpoint = hf_hub_download(
repo_id="xinlei55555/ViTPain",
filename="vit… -
Upload ViTPain pretrained checkpoint
-
initial commit
