"I used the AI to destroy the AI" - Thanos if he was a data scientist
GAUNTLET is an explainable model for detecting AI-generated images. By examining the finer characteristics of these pictures (such as noise patterns), it can distinguish AI-generated images from human-crafted ones. In addition, it outputs a heatmap which highlights the features that activated the model.
Our aim is not just to provide a tool for people to use, but also to educate them into learning the characteristics of an AI generated image, so that they can detect and avoid malicious content without the necessity of this app. In a sense, the app should train the user so that it won't rely on this app. Kinda crazy if you think about it.
You can click the following image to watch a video demo of GAUNTLET:
More details can be found in docs/report.pdf
or in the notebooks.
To use the Notebooks and the WebApp, you'll need to first create python environment and install the required packages using pip:
python -m venv .venv
- On Linux:
source .venv/bin/activate
2b. On Windows:.venv/Scripts/activate
pip install -r requirements.txt
Before running the notebooks, place the respective datasets folders in the data folder:
- CIFAKE contents into CIFAKE folder
- RealVsAiArt contents into RealVsAiArt folder
To run the webapp, use the ./start_webapp
script. If for some reason it doesn't work, you can instead use the following commands:
python webapp/server.py
cd webapp/frontend
npm install
npm start