Improve model card

#1
by nielsr HF Staff - opened
Files changed (1) hide show
  1. README.md +69 -3
README.md CHANGED
@@ -1,3 +1,69 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ pipeline_tag: image-to-image
4
+ ---
5
+
6
+ # VeraRetouch: A Lightweight Fully Differentiable Framework for Multi-Task Reasoning Photo Retouching
7
+
8
+ VeraRetouch is a lightweight and fully differentiable framework for multi-task reasoning photo retouching. It utilizes a 0.5B Vision-Language Model (VLM) to analyze image defects and formulate plans, which are then executed by a custom differentiable Retouch Renderer.
9
+
10
+ [[Paper](https://huggingface.co/papers/2604.27375)] [[Project Page](https://apollo-yi.github.io/VeraRetouch/)] [[GitHub](https://github.com/OpenVeraTeam/VeraRetouch)]
11
+
12
+ ## Overview
13
+ Existing photo retouching approaches often rely on non-differentiable external software, creating optimization barriers. VeraRetouch overcomes this with a fully differentiable Retouch Renderer, enabling direct end-to-end pixel-level training. It supports several modes:
14
+ - **Auto Mode:** Analyzes image defects and enhances them automatically.
15
+ - **Style Mode:** Retouches images based on user-provided text prompts.
16
+ - **Param Mode:** Executes retouching based on specific operator parameters.
17
+
18
+ ## Usage
19
+
20
+ ### Installation
21
+ ```bash
22
+ # Clone the repository
23
+ git clone https://github.com/OpenVeraTeam/VeraRetouch.git
24
+ cd VeraRetouch
25
+
26
+ # Create and activate conda environment
27
+ conda create -n vera-retouch python=3.10
28
+ conda activate vera-retouch
29
+ pip install -r requirements.txt
30
+ ```
31
+
32
+ ### Inference Examples
33
+ The model supports three inference modes. First, download the weights and place them in the `./checkpoints` directory.
34
+
35
+ **Auto Retouch Mode:**
36
+ ```bash
37
+ python inference.py --mode auto \
38
+ --model-path ./checkpoints/VeraRetouch \
39
+ --img_paths ./data_samples/input/sample_flower.jpg \
40
+ --save_dir ./data_samples/output/
41
+ ```
42
+
43
+ **Style Retouch Mode:**
44
+ ```bash
45
+ python inference.py --mode style \
46
+ --prompt "I want a dreamy bright pink style." \
47
+ --model-path ./checkpoints/VeraRetouch \
48
+ --img_paths ./data_samples/input/sample_flower.jpg \
49
+ --save_dir ./data_samples/output/
50
+ ```
51
+
52
+ **Param Retouch Mode:**
53
+ ```bash
54
+ python inference.py --mode style \
55
+ --instruction_path ./data_samples/param.json \
56
+ --model-path ./checkpoints/VeraRetouch \
57
+ --img_paths ./data_samples/input/sample_flower.jpg \
58
+ --save_dir ./data_samples/output/
59
+ ```
60
+
61
+ ## Citation
62
+ ```bibtex
63
+ @article{guo2026veraretouch,
64
+ title={VeraRetouch: A Lightweight Fully Differentiable Framework for Multi-Task Reasoning Photo Retouching},
65
+ author={Guo, Yihong and Lyu, Youwei and Tang, Jiajun and Zhou, Yizhuo and Wang, Hongliang and Chen, Jinwei and Zou, Changqing and Fan, Qingnan},
66
+ journal={arXiv preprint arXiv:2604.27375},
67
+ year={2026}
68
+ }
69
+ ```