LVFace Model
Browse files
README.md
CHANGED
|
@@ -39,8 +39,6 @@ This is the official PyTorch implementation for the inference of LVFace. Drawing
|
|
| 39 |
<!-- </h5> -->
|
| 40 |
</div>
|
| 41 |
|
| 42 |
-
<br>
|
| 43 |
-
|
| 44 |
<div align="center">
|
| 45 |
<img src="docs/LVFace.png" alt="LVFace η€ΊζεΎ" width="80%">
|
| 46 |
</div>
|
|
@@ -53,7 +51,6 @@ This is the official PyTorch implementation for the inference of LVFace. Drawing
|
|
| 53 |
- πππ LVFace is accepted by **ICCV 2025**. (July, 2025 UTC)
|
| 54 |
- π₯π₯π₯ We have updated the arXiv report of LVFace. Please click [here](https://arxiv.org/abs/2501.13420v2) to view it. (March, 2025 UTC)
|
| 55 |
- πππ LVFace secured **1st place** in the ICCV 2021 Masked Face Recognition (MFR)-Ongoing Challenge (academic track). (December, 2024 UTC)
|
| 56 |
-
|
| 57 |
<div align="center">
|
| 58 |
<img src="docs/MFR.png" alt="LVFace η€ΊζεΎ" width="30%">
|
| 59 |
</div>
|
|
@@ -73,13 +70,11 @@ Install all dependencies with a single command: pip install -r requirements.txt
|
|
| 73 |
|
| 74 |
## Datasets
|
| 75 |
Test datasets for inference validation can be downloaded from the following sources:
|
| 76 |
-
|
| 77 |
- **IJB-C & IJB-B**: [Google Drive](https://drive.google.com/file/d/1aC4zf2Bn0xCVH_ZtEuQipR2JvRb1bf8o/view)
|
| 78 |
- **MFR-Ongoing**: [Challenge Page](https://insightface.ai/mfr_ongoing)
|
| 79 |
|
| 80 |
## LVFace Pretrained Models
|
| 81 |
Pretrained model weights for inference are available below in both ONNX and PyTorch (.pt) formats:
|
| 82 |
-
|
| 83 |
| Training Data | Model | IJB-C(1e-6) | IJB-C(1e-5) | IJB-C(1e-4) | IJB-B(1e-4) | Download |
|
| 84 |
|---------------|---------------|-------------|-------------|-------------|-------------|---------------|
|
| 85 |
| Glint360K | LVFace-T | 88.53 | 95.63 | 96.67 | 95.41 | [HuggingFace](https://huggingface.co/bytedance-research/LVFace/tree/main/LVFace-T_Glint360K) |
|
|
@@ -92,7 +87,7 @@ Pretrained model weights for inference are available below in both ONNX and PyTo
|
|
| 92 |
## Step-by-Step Usage Guide
|
| 93 |
|
| 94 |
**1. Installation & Environment Setup**
|
| 95 |
-
|
| 96 |
First, clone the repository and navigate to the project directory:
|
| 97 |
```bash
|
| 98 |
# Clone the LVFace repository
|
|
@@ -135,17 +130,13 @@ Pretrained model weights for inference are available below in both ONNX and PyTo
|
|
| 135 |
**Note**: The `LVFaceONNXInferencer` class is defined in `inference_onnx.py`, which handles ONNX model loading, image preprocessing, feature extraction, and similarity calculation in a unified interface. Ensure the model path and image paths are correctly specified before running.
|
| 136 |
|
| 137 |
|
| 138 |
-
|
| 139 |
## Model Evaluation
|
| 140 |
|
| 141 |
**Evaluation Steps**
|
| 142 |
1. Modify the test dataset path (e.g., IJB-C, IJB-B) in the corresponding evaluation script (`eval_ijbc.py`).
|
| 143 |
-
|
| 144 |
2. Run the evaluation with pretrained model weights using the commands below:
|
| 145 |
|
| 146 |
-
|
| 147 |
**Evaluation Commands**
|
| 148 |
-
|
| 149 |
```bash
|
| 150 |
python eval_ijbc.py \
|
| 151 |
--model-prefix path/to/LVFace-B_Glint360K.pt \
|
|
@@ -156,7 +147,6 @@ python eval_ijbc.py \
|
|
| 156 |
|
| 157 |
## License
|
| 158 |
The code of LVFace is released under the MIT License. There is no limitation for both academic and commercial usage.
|
| 159 |
-
|
| 160 |
The models downloaded from our repo follow the above license policy (which is for non-commercial research purposes only).
|
| 161 |
|
| 162 |
|
|
@@ -173,5 +163,4 @@ If you find this work useful, please cite our paper and give us a star β:
|
|
| 173 |
|
| 174 |
## Acknowledgments
|
| 175 |
We sincerely thank **Professor Deng Jiankang** for his valuable guidance and insights throughout the research.
|
| 176 |
-
|
| 177 |
We also appreciate the [InsightFace](https://github.com/deepinsight/insightface/tree/master/recognition/arcface_torch) for their excellent and research support.
|
|
|
|
| 39 |
<!-- </h5> -->
|
| 40 |
</div>
|
| 41 |
|
|
|
|
|
|
|
| 42 |
<div align="center">
|
| 43 |
<img src="docs/LVFace.png" alt="LVFace η€ΊζεΎ" width="80%">
|
| 44 |
</div>
|
|
|
|
| 51 |
- πππ LVFace is accepted by **ICCV 2025**. (July, 2025 UTC)
|
| 52 |
- π₯π₯π₯ We have updated the arXiv report of LVFace. Please click [here](https://arxiv.org/abs/2501.13420v2) to view it. (March, 2025 UTC)
|
| 53 |
- πππ LVFace secured **1st place** in the ICCV 2021 Masked Face Recognition (MFR)-Ongoing Challenge (academic track). (December, 2024 UTC)
|
|
|
|
| 54 |
<div align="center">
|
| 55 |
<img src="docs/MFR.png" alt="LVFace η€ΊζεΎ" width="30%">
|
| 56 |
</div>
|
|
|
|
| 70 |
|
| 71 |
## Datasets
|
| 72 |
Test datasets for inference validation can be downloaded from the following sources:
|
|
|
|
| 73 |
- **IJB-C & IJB-B**: [Google Drive](https://drive.google.com/file/d/1aC4zf2Bn0xCVH_ZtEuQipR2JvRb1bf8o/view)
|
| 74 |
- **MFR-Ongoing**: [Challenge Page](https://insightface.ai/mfr_ongoing)
|
| 75 |
|
| 76 |
## LVFace Pretrained Models
|
| 77 |
Pretrained model weights for inference are available below in both ONNX and PyTorch (.pt) formats:
|
|
|
|
| 78 |
| Training Data | Model | IJB-C(1e-6) | IJB-C(1e-5) | IJB-C(1e-4) | IJB-B(1e-4) | Download |
|
| 79 |
|---------------|---------------|-------------|-------------|-------------|-------------|---------------|
|
| 80 |
| Glint360K | LVFace-T | 88.53 | 95.63 | 96.67 | 95.41 | [HuggingFace](https://huggingface.co/bytedance-research/LVFace/tree/main/LVFace-T_Glint360K) |
|
|
|
|
| 87 |
## Step-by-Step Usage Guide
|
| 88 |
|
| 89 |
**1. Installation & Environment Setup**
|
| 90 |
+
|
| 91 |
First, clone the repository and navigate to the project directory:
|
| 92 |
```bash
|
| 93 |
# Clone the LVFace repository
|
|
|
|
| 130 |
**Note**: The `LVFaceONNXInferencer` class is defined in `inference_onnx.py`, which handles ONNX model loading, image preprocessing, feature extraction, and similarity calculation in a unified interface. Ensure the model path and image paths are correctly specified before running.
|
| 131 |
|
| 132 |
|
|
|
|
| 133 |
## Model Evaluation
|
| 134 |
|
| 135 |
**Evaluation Steps**
|
| 136 |
1. Modify the test dataset path (e.g., IJB-C, IJB-B) in the corresponding evaluation script (`eval_ijbc.py`).
|
|
|
|
| 137 |
2. Run the evaluation with pretrained model weights using the commands below:
|
| 138 |
|
|
|
|
| 139 |
**Evaluation Commands**
|
|
|
|
| 140 |
```bash
|
| 141 |
python eval_ijbc.py \
|
| 142 |
--model-prefix path/to/LVFace-B_Glint360K.pt \
|
|
|
|
| 147 |
|
| 148 |
## License
|
| 149 |
The code of LVFace is released under the MIT License. There is no limitation for both academic and commercial usage.
|
|
|
|
| 150 |
The models downloaded from our repo follow the above license policy (which is for non-commercial research purposes only).
|
| 151 |
|
| 152 |
|
|
|
|
| 163 |
|
| 164 |
## Acknowledgments
|
| 165 |
We sincerely thank **Professor Deng Jiankang** for his valuable guidance and insights throughout the research.
|
|
|
|
| 166 |
We also appreciate the [InsightFace](https://github.com/deepinsight/insightface/tree/master/recognition/arcface_torch) for their excellent and research support.
|