README.md

metadata

license: apache-2.0
title: Heatmap
sdk: streamlit
colorFrom: gray
colorTo: gray
pinned: true
sdk_version: 1.43.2
app_file: streamlit_app.py

Deepfake Detection System with Heatmap Visualization

This project implements a deepfake detection system that processes images through multiple stages:

1. Verification Module: Uses Nvidia AI model for image comparison - Calculates SMI (Structural Matching Index) - Generates difference images

   Applies threshold visualization

   • Adds bounding boxes around detected areas

2. Labeling System: Labels detected areas based on threat level

3. Heatmap Visualization: Creates heatmaps highlighting high-threat areas

Project Structure

• deepfake_detector.py: Core detection functionality using Nvidia AI model
• image_processor.py: Image processing utilities
• labeling.py: Threat labeling system
• heatmap_generator.py: Heatmap visualization tools
• main.py: Main application entry point
• requirements.txt: Project dependencies

Setup and Installation

pip install -r requirements.txt

Usage

python main.py --input_dir /path/to/images --output_dir /path/to/output