ZuhairQureshi commited on
Commit
310fbb6
·
verified ·
1 Parent(s): 90ae8e2

Upload ViT Base artifacts (weights + configs + preprocess)

Browse files
README.md CHANGED
@@ -1,3 +1,18 @@
1
- ---
2
- license: mit
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
+ }