Eurong2 commited on
Commit
4356a92
·
verified ·
1 Parent(s): f51bdbb

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ training_state/200000.state filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ pipeline_tag: image-to-image
4
+ tags:
5
+ - watermarking
6
+ - image-self-recovery
7
+ - tamper-localization
8
+ - invertible-neural-network
9
+ ---
10
+
11
+ # ReImage: Robust Image Self-Recovery against Tampering using Watermark Generation with Pixel Shuffling
12
+
13
+ Official pretrained weights for **ReImage**, a neural watermarking-based image self-recovery framework that embeds a shuffled version of the target image into itself as a watermark, enabling tamper localization and recovery of the original content.
14
+
15
+ - 📄 Paper: [arXiv:2511.22936](https://arxiv.org/abs/2511.22936)
16
+ - 🌐 Project page: https://eurominyoung186.github.io/ReImage/
17
+ - 💻 Code: https://github.com/EuroMinyoung186/ReImage
18
+
19
+ ## Files
20
+
21
+ | File | Description |
22
+ |---|---|
23
+ | `models/200000_G.pth` | Network weights at iteration 200k (use this for inference) |
24
+ | `training_state/200000.state` | Optimizer/scheduler state at iteration 200k (only needed to resume training) |
25
+
26
+ ## Download
27
+
28
+ ```bash
29
+ pip install -U huggingface_hub
30
+ hf download Eurong2/ReImage --local-dir ../experiments/invertible_pipeline
31
+ ```
32
+
33
+ The repo mirrors the directory layout expected by the test configs (an `experiments/` directory next to the code repository root):
34
+
35
+ ```
36
+ ../experiments/invertible_pipeline/models/200000_G.pth
37
+ ../experiments/invertible_pipeline/training_state/200000.state
38
+ ```
39
+
40
+ Then run inference following the instructions in the code repository:
41
+
42
+ ```bash
43
+ python test2.py -opt options/test/val_LF-VSN_1video_celebA.yml # predicted tamper masks
44
+ python test.py -opt options/test/photoshop.yml # ground-truth masks
45
+ ```
46
+
47
+ ## Citation
48
+
49
+ ```bibtex
50
+ @article{kim2025reimage,
51
+ title = "Robust Image Self-Recovery against Tampering using Watermark Generation with Pixel Shuffling",
52
+ author = "Kim, Minyoung and Seo, Paul Hongsuck",
53
+ journal = "arXiv preprint arXiv:2511.22936",
54
+ year = "2025"
55
+ }
56
+ ```
models/200000_G.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c057649c1ca8eb81838371128ccae8f309d16c59650cf17f93d138c37db1b59d
3
+ size 40103611
training_state/200000.state ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06314168a086c895016635eb877ac8023373c894731eaf3af299f44ade94bcbf
3
+ size 80152669