Upload ViT Base artifacts (weights + configs + preprocess)
Browse files- README.md +18 -3
- config.json +16 -0
- deepfake_vit_finetuned_wildfake.pth +3 -0
- label_map.json +4 -0
- preprocess.json +16 -0
README.md
CHANGED
|
@@ -1,3 +1,18 @@
|
|
| 1 |
-
---
|
| 2 |
-
license:
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- vision-transformer
|
| 5 |
+
- deepfake-detection
|
| 6 |
+
- image-classification
|
| 7 |
+
---
|
| 8 |
+
|
| 9 |
+
# ViT Base – DeepFakeDetector
|
| 10 |
+
|
| 11 |
+
Vision Transformer (ViT-B/16) fine-tuned for binary deepfake detection.
|
| 12 |
+
|
| 13 |
+
## Labels
|
| 14 |
+
- 0 = real
|
| 15 |
+
- 1 = fake
|
| 16 |
+
|
| 17 |
+
## Output
|
| 18 |
+
- prob_fake ∈ [0,1] (softmax or sigmoid)
|
config.json
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "vit-base",
|
| 3 |
+
"framework": "pytorch",
|
| 4 |
+
"arch": "vit_base_patch16_224",
|
| 5 |
+
"patch_size": 16,
|
| 6 |
+
"hidden_dim": 768,
|
| 7 |
+
"num_layers": 12,
|
| 8 |
+
"num_heads": 12,
|
| 9 |
+
"num_classes": 2,
|
| 10 |
+
"threshold": 0.5,
|
| 11 |
+
"labels": {
|
| 12 |
+
"0": "real",
|
| 13 |
+
"1": "fake"
|
| 14 |
+
},
|
| 15 |
+
"notes": "ViT Base transfer learning model for deepfake detection"
|
| 16 |
+
}
|
deepfake_vit_finetuned_wildfake.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6da70e21817a0c591e09a11c0a3ebaac649671392f40721a50baefbf83e46cdb
|
| 3 |
+
size 1034527103
|
label_map.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"0": "real",
|
| 3 |
+
"1": "fake"
|
| 4 |
+
}
|
preprocess.json
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"input_size": 224,
|
| 3 |
+
"interpolation": "bicubic",
|
| 4 |
+
"normalize": {
|
| 5 |
+
"mean": [
|
| 6 |
+
0.485,
|
| 7 |
+
0.456,
|
| 8 |
+
0.406
|
| 9 |
+
],
|
| 10 |
+
"std": [
|
| 11 |
+
0.229,
|
| 12 |
+
0.224,
|
| 13 |
+
0.225
|
| 14 |
+
]
|
| 15 |
+
}
|
| 16 |
+
}
|