ai-image-detector / config.json
fatihunal's picture
Duplicate from wkaandemir/ai-image-detector
b664492
Raw
History Blame Contribute Delete
1.12 kB
{
"model_type": "ai-image-detector",
"architectures": ["ViTForImageClassification"],
"backbone": "vit_base_patch16_clip_224.openai",
"backbone_type": "clip_vit_base_patch16",
"patch_size": 16,
"image_size": 256,
"num_classes": 1,
"in_channels": 3,
"task": "image-classification",
"score_semantics": "sigmoid output is p(real); 0=fake, 1=real",
"decision_rule": "fake if p(real) < fake_threshold; real if p(real) >= real_threshold; otherwise uncertain",
"fake_threshold": 0.91,
"real_threshold": 0.93,
"temperature": 0.594889223575592,
"calibration_method": "temperature_nll",
"threshold_source": "dedicated_calibration_split",
"normalization_mean": [0.481, 0.458, 0.408],
"normalization_std": [0.269, 0.261, 0.276],
"training": {
"finetune_mode": "lora",
"lora_r": 16,
"lora_alpha": 32,
"lora_dropout": 0.05,
"lora_targets": ["qkv", "attn.proj"],
"augmentation_profile": "forensic",
"epochs_run": 12,
"epochs_requested": 20,
"early_stopped": true,
"seed": 42
},
"id2label": {"0": "fake", "1": "real"},
"label2id": {"fake": 0, "real": 1}
}