Spaces:
Sleeping
Sleeping
File size: 1,187 Bytes
4a81555 56044f2 433addc 4a81555 4882e62 9e629a3 433addc 9e629a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
---
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
```bash
pip install -r requirements.txt
```
## Usage
```bash
python main.py --input_dir /path/to/images --output_dir /path/to/output
``` |