Upload CNN transfer artifacts (model + preprocess + config)
Browse files- README.md +23 -3
- config.json +13 -0
- label_map.json +4 -0
- model.pth +3 -0
- preprocess.json +29 -0
README.md
CHANGED
|
@@ -1,3 +1,23 @@
|
|
| 1 |
-
---
|
| 2 |
-
license:
|
| 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 |
+
}
|