StableGuard

StableGuard is an AI-powered application designed to detect whether a given image or video is AI-generated, such as those created by stable diffusion models and other AI tools.

Note: The detection results may not always be accurate. AI-generated content detection is an evolving field, and false positives or false negatives are possible. Always verify the results with additional methods or human judgment when necessary.

Features

• Detects AI-generated images and videos
• Uses pre-trained machine learning models for detection
• Provides confidence and accuracy metrics for classification
• Supports both image and video processing

Requirements

• Python 3.11+
• PyTorch
• OpenCV
• NumPy
• Pillow
• Transformers
• Accelerate
• Datasets
• Gradio
• Huggingface Models

Installation

1. Clone the repository:

   git clone https://github.com/muhammad-fiaz/StableGuard.git
   cd StableGuard

2. Install the required packages:

   pip install -r requirements.txt

Usage

1. To detect AI-generated images:

   python detect.py --image path/to/image.jpg

2. To detect AI-generated videos:

   python detect.py --video path/to/video.mp4

3. To detect with GUI

   python detect.py --gui

License

This project is licensed under the Apache License 2.0 - see the LICENSE file for details.