File size: 1,198 Bytes
154d677 4fd5a38 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
---
license: mit
library_name: transformers
tags:
- pytorch
- image-classification
---
# Deepfake Detection that Generalizes Across Benchmarks (WACV 2026)
[](https://arxiv.org/abs/2508.06248)
[](https://github.com/yermandy/GenD)
This is the GenD (PE) model from Tab. 2 in the paper.
## Set up environment
``` bash
conda create --name GenD python=3.12 uv
conda activate GenD
uv pip install -r requirements.txt
```
## Usage
``` python
import requests
import torch
from PIL import Image
from src.hf.modeling_gend import GenD
model = GenD.from_pretrained("yermandy/GenD_PE_L")
urls = [
"https://github.com/yermandy/deepfake-detection/blob/main/datasets/FF/DF/000_003/000.png?raw=true",
"https://github.com/yermandy/deepfake-detection/blob/main/datasets/FF/real/000/000.png?raw=true",
]
images = [Image.open(requests.get(url, stream=True).raw) for url in urls]
tensors = torch.stack([model.feature_extractor.preprocess(img) for img in images])
logits = model(tensors)
probs = logits.softmax(dim=-1)
print(probs)
```
|