CoolFace
Modelpublic

alianassmaaa/multimodal-deepfake-detector

sourceHugging Faceapache-2.0updated 5mo agoView on Hugging Face
0likes44downloads
Model Card

🕵️ Multimodal Deepfake Detection Model

Modèle d'ensemble multimodal pour détecter les contenus générés par IA (images, vidéos, texte) avec explicabilité GradCAM.

🏗️ Architecture

Visual Branch    EfficientNet-B0/B4  ──┐
                                       ├──► Fusion pondérée ──► Confidence [0-1]
Text Branch      RoBERTa-base         ──┘
                    ↓
               GradCAM Heatmap (explicabilité)

🚀 Utilisation Rapide

python
from inference import load_model, classify_image, classify_text, classify_multimodal

model, config = load_model('multimodal_ensemble.pt', device='cuda')

# Image + GradCAM
result = classify_image(model, 'face.jpg', return_gradcam=True)
print(f"{result['prediction']} — confidence: {result['confidence']:.2%}")
# result['gradcam'] → heatmap (224, 224)

# Texte
result = classify_text(model, 'Cet essai analyse les impacts...')
print(f"{result['prediction']} — confidence: {result['confidence']:.2%}")

# Multimodal (image + texte)
result = classify_multimodal(model, image_path_or_pil='face.jpg', text='Caption...')
print(f"Fusion: {result['prediction']} — poids: {result['fusion_weights']}")

📊 Jeux de Données

📁 Fichiers

FichierDescription
model.pyArchitecture complète
preprocessing.pyPipeline de données
inference.pyAPI d'inférence
train.py / train_optimised.pyScripts d'entraînement
multimodal_ensemble.ptCheckpoint principal
gradcam_examples/Visualisations d'explicabilité

🔗 Space de Démo

alianassmaaa/multimodal-deepfake-space

📄 Licence

Apache-2.0