Skip to content

Mr-Infect/image-forensic-toolkit

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🛡️ Image Forensic Toolkit

Advanced digital image forensics toolkit for analyzing evidence, detecting AI generation, and identifying image manipulations. Built with Streamlit and modern AI models (CLIP, ViT).

🚀 Features

  • 🖼️ Similarity Search: Find visually and semantically similar images in the database via image upload or text keywords (using CLIP).
  • 🤖 AI Generation Detection: Identify if an image is AI-generated (GAN/Diffusion) with confidence scores.
  • 🛠️ Manipulation Analysis: Perform Error Level Analysis (ELA) and statistical attribute analysis to detect tampering.
  • 📄 Metadata & Steganography: Extract deep EXIF data and detect hidden scripts or high-entropy anomalies.
  • 👤 Privacy Blur: Automatically detect and blur human subjects in reports using high-recall face detection.
  • 📥 Reporting: Export all forensic findings as structured JSON or human-readable TXT files.

🛠️ Installation & Setup

1. Prerequisites

Ensure you have Python 3.10+ installed on your system.

2. Clone the Repository

git clone https://github.com/Mr-Infect/image-forensic-toolkit.git
cd image-forensic-toolkit

3. Set Up Environment

It is recommended to use a virtual environment:

python3 -m venv venv
source venv/bin/activate  # On Windows: venv\Scripts\activate

4. Install Dependencies

pip install -r requirements.txt

5. Download AI Models

Before running the application, you must cache the necessary AI models. You will need a Hugging Face token.

python3 download_models.py

(Progress will be displayed in the terminal during download)

🖥️ Usage

Once the models are downloaded, launch the Streamlit application:

streamlit run app.py

🗃️ Indexing the Database

On the first run, navigate to the About section in the app and click "Re-index Database Images". This will process the images in the images/ folder for similarity search.

📂 Project Structure

  • app.py: Main Streamlit application.
  • modules/: Core forensic logic components.
  • images/: Centralized folder for image database.
  • download_models.py: Utility script for pre-caching models.

Developed for Image Forensic & Evidence Analysis.

About

Advanced digital image forensics toolkit for analyzing evidence, detecting AI generation, and identifying image manipulations. Built with Streamlit and modern AI models (CLIP, ViT).

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages