nebula commited on
Commit
765f09a
·
verified ·
1 Parent(s): f2da512

Add model card

Browse files
Files changed (1) hide show
  1. README.md +55 -0
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: pytorch
3
+ tags:
4
+ - opensdi
5
+ - maskclip
6
+ - diffusion-detection
7
+ - image-forensics
8
+ - forgery-localization
9
+ - pytorch
10
+ datasets:
11
+ - nebula/OpenSDI_train
12
+ - nebula/OpenSDI_test
13
+ ---
14
+
15
+ # MaskCLIP Weights for OpenSDI
16
+
17
+ This repository hosts model checkpoints for **OpenSDI: Spotting Diffusion-Generated Images in the Open World**.
18
+
19
+ ## Links
20
+
21
+ - Model weights: https://huggingface.co/nebula/MaskCLIP-weights/tree/main
22
+ - Code: https://github.com/iamwangyabin/OpenSDI
23
+ - Project page: https://iamwangyabin.github.io/OpenSDI/
24
+ - Paper: https://arxiv.org/abs/2503.19653
25
+ - Training dataset: https://huggingface.co/datasets/nebula/OpenSDI_train
26
+ - Testing dataset: https://huggingface.co/datasets/nebula/OpenSDI_test
27
+
28
+ ## Checkpoints
29
+
30
+ The `Files and versions` tab contains `.pth` checkpoints for MaskCLIP and related OpenSDI baselines. For MaskCLIP evaluation, download one of the `MaskCLIP_sd15_*.pth` checkpoints and use it with the OpenSDI codebase.
31
+
32
+ Example:
33
+
34
+ ```bash
35
+ hf download nebula/MaskCLIP-weights MaskCLIP_sd15_20241103_17_45_16.pth --local-dir weights
36
+ ```
37
+
38
+ Then set `--checkpoint_path` in `test.sh` to the downloaded checkpoint path, for example:
39
+
40
+ ```bash
41
+ --checkpoint_path "weights/MaskCLIP_sd15_20241103_17_45_16.pth"
42
+ ```
43
+
44
+ ## Citation
45
+
46
+ If you find OpenSDI useful for your research and applications, please cite:
47
+
48
+ ```bibtex
49
+ @InProceedings{wang2025opensdi,
50
+ author={Wang, Yabin and Huang, Zhiwu and Hong, Xiaopeng},
51
+ title={OpenSDI: Spotting Diffusion-Generated Images in the Open World},
52
+ booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
53
+ year={2025}
54
+ }
55
+ ```