zhangjinyang commited on
Commit
8574665
·
verified ·
1 Parent(s): 28456a2

Add model card assets

Browse files
.gitattributes CHANGED
@@ -33,3 +33,6 @@ 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
+ assets/noref-aware_vis.png filter=lfs diff=lfs merge=lfs -text
37
+ assets/ref-aware_vis.png filter=lfs diff=lfs merge=lfs -text
38
+ assets/teaser_example.png filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -1,3 +1,112 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ base_model: black-forest-labs/FLUX.2-klein-base-4B
4
+ tags:
5
+ - face-restoration
6
+ - blind-face-restoration
7
+ - reference-guided-restoration
8
+ - flux
9
+ - lora
10
+ - computer-vision
11
+ pipeline_tag: image-to-image
12
+ ---
13
+
14
+ # IConFace LoRA Adapter
15
+
16
+ This repository hosts the released LoRA adapter for **IConFace: Identity-Structure
17
+ Asymmetric Conditioning for Unified Reference-Aware Face Restoration**.
18
+
19
+ - Project page: https://cosmicrealm.github.io/IConFace/
20
+ - arXiv: https://arxiv.org/abs/2605.02814
21
+ - Code: https://github.com/cosmicrealm/IConFace
22
+ - Model weights: https://huggingface.co/zhangjinyang/IConFace
23
+
24
+ ## Visual Results
25
+
26
+ <p align="center">
27
+ <img src="assets/teaser_example.png" alt="IConFace detail restoration example" width="900">
28
+ </p>
29
+
30
+ ### Reference-Aware Restoration
31
+
32
+ <p align="center">
33
+ <img src="assets/ref-aware_vis.png" alt="Reference-aware visual results" width="900">
34
+ </p>
35
+
36
+ ### No-Reference Restoration
37
+
38
+ <p align="center">
39
+ <img src="assets/noref-aware_vis.png" alt="No-reference visual results" width="900">
40
+ </p>
41
+
42
+ More visual comparisons are available on the project page.
43
+
44
+ ## Files
45
+
46
+ ```text
47
+ iconface_lora.safetensors
48
+ ```
49
+
50
+ This adapter is used with:
51
+
52
+ - Base model: `black-forest-labs/FLUX.2-klein-base-4B`
53
+ - LoRA target preset: `flux2_klein`
54
+ - LoRA rank: 16
55
+ - Expected path in the code release: `pretrained_models/iconface_lora.safetensors`
56
+
57
+ ## Usage
58
+
59
+ Use the GitHub code release:
60
+
61
+ ```bash
62
+ git clone https://github.com/cosmicrealm/IConFace.git
63
+ cd IConFace
64
+ pip install -r requirements.txt
65
+ bash scripts/download_models.sh
66
+ ```
67
+
68
+ No-reference inference:
69
+
70
+ ```bash
71
+ python scripts/infer_no_ref.py \
72
+ --model_dir pretrained_models/FLUX.2-klein-base-4B \
73
+ --lora_path pretrained_models/iconface_lora.safetensors \
74
+ --adaface_model_path pretrained_models/adaface_ir50_ms1mv2.ckpt \
75
+ --condition_image examples/lq.png \
76
+ --output_dir outputs/no_ref_demo
77
+ ```
78
+
79
+ Reference-aware inference:
80
+
81
+ ```bash
82
+ python scripts/infer_with_ref.py \
83
+ --model_dir pretrained_models/FLUX.2-klein-base-4B \
84
+ --lora_path pretrained_models/iconface_lora.safetensors \
85
+ --adaface_model_path pretrained_models/adaface_ir50_ms1mv2.ckpt \
86
+ --condition_image examples/lq.png \
87
+ --reference_image examples/ref1.png \
88
+ --reference_image examples/ref2.png \
89
+ --reference_image examples/ref3.png \
90
+ --output_dir outputs/ref_demo
91
+ ```
92
+
93
+ ## Citation
94
+
95
+ ```bibtex
96
+ @misc{niu2026iconface,
97
+ title={IConFace: Identity-Structure Asymmetric Conditioning for Unified Reference-Aware Face Restoration},
98
+ author={Niu, Axi and Zhang, Jinyang and Qing, Senyan},
99
+ year={2026},
100
+ eprint={2605.02814},
101
+ archivePrefix={arXiv},
102
+ primaryClass={cs.CV},
103
+ doi={10.48550/arXiv.2605.02814},
104
+ url={https://arxiv.org/abs/2605.02814}
105
+ }
106
+ ```
107
+
108
+ ## License Notes
109
+
110
+ This adapter depends on the FLUX.2-Klein-base-4B base model. Users must comply
111
+ with the license and access terms of the base model and all external identity
112
+ models used by the code release.
assets/noref-aware_vis.png ADDED

Git LFS Details

  • SHA256: 4ccbba3f9715de23217e9eb05043402ea93719ea11524e2bf03a4cf1305eab04
  • Pointer size: 132 Bytes
  • Size of remote file: 8.27 MB
assets/ref-aware_vis.png ADDED

Git LFS Details

  • SHA256: 406330db1d45b44a4d4ad4b263f30dfb330720cb5b4b8efd18269d975db496ac
  • Pointer size: 132 Bytes
  • Size of remote file: 2.64 MB
assets/teaser_example.png ADDED

Git LFS Details

  • SHA256: 787f403838088d686b73bde464c5e0c01c2a21131d71ce85684f0e152f1c7d20
  • Pointer size: 132 Bytes
  • Size of remote file: 2.6 MB