YichiCS / DNF

Code repository for the paper --- DNF

Diffusion-Noise-Feature-Accurate-and-Fast-Generated-Image-Detection

Yichi Zhang and Xiaogang Xu

Code repository for the paper: Diffusion Noise Feature: Accurate and Fast Generated Image Detection.

Baseline

The code is based on CNNDetction

Model Preparation

The checkpoints can be download from Huggingface.

Download the LSUN Bedroom pretrained DDIM from here. Place it in ./weights/diffusion/ .

Dataset Preparation

The testsets can be download from Huggingface. Download the DiffusionForensics from DIRE.

Please refer to CNNDetction for the storage path of the dataset.

Transform Image to DNF

python compute_dnf.py

Training

python train.py 

Evaluation

python eval.py 

Please refer to ./options for variables that determine the program’s execution.

Citation

@article{zhang2023diffusion,
  title={Diffusion noise feature: Accurate and fast generated image detection},
  author={Zhang, Yichi and Xu, Xiaogang},
  journal={arXiv preprint arXiv:2312.02625},
  year={2023}
}