Artifact Purification Network

This is the official codes for the paper "Artifact feature purification for cross-domain detection of AI-generated images", published in Computer Vision and Image Understanding.

How to use

1. Install requirements

conda create -n APN python==3.7
conda activate APN
pip install -r requirements.txt

2. Download pretrained models Google Drive

3. Run code

sh scripts/eval.sh

4. If you want to train the model

sh scripts/run.sh

Results on GenImage