ash12321's picture
Upload folder using huggingface_hub
fcd620e verified
# DeepFake Detector V14 🎯
**Weighted ensemble of V13 models**
## Overview
V14 is a weighted ensemble that combines the three V13 models:
- Model 13.1 (ConvNeXt-Large): Weight 0.25
- Model 13.2 (ViT-Large): Weight 0.35
- Model 13.3 (Swin-Large): Weight 0.40
**Expected F1 Score**: 0.9361
## Model Files
- `model_1.safetensors` - ConvNeXt-Large (788 MB)
- `model_2.safetensors` - ViT-Large (1220 MB)
- `model_3.safetensors` - Swin-Large (783 MB)
- `ensemble.pth` - Ensemble wrapper weights
- `config.json` - Configuration
- `inference_example.py` - Usage example
## Quick Start
```python
# See inference_example.py for complete code
# Load all 3 models, run predictions, compute weighted average
```
## Performance
| Model | Backbone | F1 Score | Ensemble Weight |
|-------|----------|----------|-----------------|
| Model 13.1 | ConvNeXt-Large | 0.8971 | 0.25 |
| Model 13.2 | ViT-Large | 0.9382 | 0.35 |
| Model 13.3 | Swin-Large | 0.9586 | 0.40 |
**Weighted Average F1**: 0.9361
## Requirements
```
torch>=2.0.0
timm>=0.9.0
torchvision>=0.15.0
safetensors
pillow
```
## Citation
```bibtex
@model{v14-deepfake-detector,
title={DeepFake Detector V14},
author={Ash},
year={2024},
publisher={Hugging Face}
}
```
## Predecessor
Built on: [`ash12321/deepfake-detector-v13`](https://huggingface.co/ash12321/deepfake-detector-v13)