alifiscool86 commited on
Commit
ab91377
·
verified ·
1 Parent(s): 85e05aa

Upload CNN transfer artifacts (model + preprocess + config)

Browse files
Files changed (5) hide show
  1. README.md +23 -3
  2. config.json +13 -0
  3. label_map.json +4 -0
  4. model.pth +3 -0
  5. preprocess.json +29 -0
README.md CHANGED
@@ -1,3 +1,23 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - image-classification
5
+ - deepfake-detection
6
+ - pytorch
7
+ ---
8
+
9
+ # CNN Transfer (DeepFakeDetector)
10
+
11
+ Binary image classifier for deepfake or AI-generated image detection.
12
+
13
+ ## Label convention (fixed)
14
+ - 0 = real
15
+ - 1 = fake
16
+
17
+ ## Preprocessing
18
+ - Resize to 224x224
19
+ - Normalize with ImageNet mean/std
20
+
21
+ ## Output
22
+ - prob_fake in [0, 1] computed as softmax(logits)[fake]
23
+ - decision threshold: 0.662
config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "cnn-transfer",
3
+ "framework": "pytorch",
4
+ "arch": "efficientnet_b0",
5
+ "num_classes": 2,
6
+ "threshold": 0.662,
7
+ "labels": {
8
+ "0": "real",
9
+ "1": "fake"
10
+ },
11
+ "exported_at_utc": "2026-02-16T23:35:43.373902Z",
12
+ "notes": "CNN transfer baseline for deepfake detection. prob_fake = softmax(logits)[fake]."
13
+ }
label_map.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "0": "real",
3
+ "1": "fake"
4
+ }
model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5671df633967e535bd7f0dc9a2fd5000cd9edd6358da0f7f0188e94580144610
3
+ size 16320083
preprocess.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "input_size": [
3
+ 224,
4
+ 224
5
+ ],
6
+ "resize": {
7
+ "enabled": true,
8
+ "size": [
9
+ 224,
10
+ 224
11
+ ]
12
+ },
13
+ "center_crop": {
14
+ "enabled": false
15
+ },
16
+ "normalize": {
17
+ "mean": [
18
+ 0.485,
19
+ 0.456,
20
+ 0.406
21
+ ],
22
+ "std": [
23
+ 0.229,
24
+ 0.224,
25
+ 0.225
26
+ ]
27
+ },
28
+ "notes": "Resize to IM_SIZE x IM_SIZE then normalize with ImageNet mean/std."
29
+ }