lime-j commited on
Commit
e6849b6
·
1 Parent(s): dd0c824

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -99
README.md CHANGED
@@ -1,99 +1,8 @@
1
- <p align="center">
2
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/Title.png?raw=true" width=95%>
3
- <p>
4
-
5
- # Reversible Decoupling Network for Single Image Reflection Removal
6
-
7
- <div align="center">
8
-
9
- [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/reversible-decoupling-network-for-single/reflection-removal-on-sir-2-objects)](https://paperswithcode.com/sota/reflection-removal-on-sir-2-objects?p=reversible-decoupling-network-for-single)
10
- [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/reversible-decoupling-network-for-single/reflection-removal-on-sir-2-wild)](https://paperswithcode.com/sota/reflection-removal-on-sir-2-wild?p=reversible-decoupling-network-for-single)
11
- [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/reversible-decoupling-network-for-single/reflection-removal-on-sir-2-postcard)](https://paperswithcode.com/sota/reflection-removal-on-sir-2-postcard?p=reversible-decoupling-network-for-single)
12
- [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/reversible-decoupling-network-for-single/reflection-removal-on-nature)](https://paperswithcode.com/sota/reflection-removal-on-nature?p=reversible-decoupling-network-for-single)
13
- [![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/reversible-decoupling-network-for-single/reflection-removal-on-real20)](https://paperswithcode.com/sota/reflection-removal-on-real20?p=reversible-decoupling-network-for-single)
14
-
15
- </div>
16
- <p align="center" style="font-size: larger;">
17
- <a href="https://arxiv.org/abs/2410.08063"> Reversible Decoupling Network for Single Image Reflection Removal</a>
18
- </p>
19
- <p align="center">
20
- <a href="https://github.com/WHiTEWoLFJ"> Hao Zhao</a> ⚔️,
21
- <a href="https://github.com/lime-j"> Mingjia Li</a> ⚔️,
22
- <a href="https://github.com/mingcv"> Qiming Hu</a>,
23
- <a href="https://sites.google.com/view/xjguo"> Xiaojie Guo</a> 🦅,
24
- <p align="center">(⚔️: equal contribution, 🦅 : corresponding author)</p>
25
- </p>
26
-
27
- <p align="center">
28
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/net.png?raw=true" width=95%>
29
- <p>
30
- **Our work is accepted by CVPR 2025! See you at the conference!**
31
- <details>
32
- <summary>Click for the Abstract of RDNet</summary>
33
- We present a Reversible Decoupling Network (RDNet), which employs a reversible encoder to secure valuable information while flexibly decoupling transmission-and-reflection-relevant features during the forward pass. Furthermore, we customize a transmission-rate-aware prompt generator to dynamically calibrate features, further boosting performance. Extensive experiments demonstrate the superiority of RDNet over existing SOTA methods on five widely-adopted benchmark datasets.
34
- </details>
35
-
36
- ## 🚀Todo
37
-
38
- - [ ] Release the Training code of RDNet.
39
-
40
- ## 🌠 Gallery
41
-
42
-
43
- <table class="center">
44
- <tr>
45
- <td><p style="text-align: center">Class Room</p></td>
46
- <td><p style="text-align: center">White Wall Chamber</p></td>
47
- </tr>
48
- <tr>
49
- <td>
50
- <div style="width: 100%; max-width: 600px; position: relative;">
51
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/Input_class.png?raw=true" style="width: 100%; height: 300px; display: block;">
52
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/Ours_class.png?raw=true" style="width: 100%; height: 300px; display: block; position: absolute; top: 0; left: 0; opacity: 0; transition: opacity 0.5s;" onmouseover="this.style.opacity=1;" onmouseout="this.style.opacity=0;">
53
- </div>
54
- </td>
55
- <td>
56
- <div style="width: 100%; max-width: 600px; position: relative;">
57
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/input_white.jpg?raw=true" style="width: 100%; height: 300px; display: block;">
58
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/Ours_white.png?raw=true" style="width: 100%; height: 300px; display: block; position: absolute; top: 0; left: 0; opacity: 0; transition: opacity 0.5s;" onmouseover="this.style.opacity=1;" onmouseout="this.style.opacity=0;">
59
- </div>
60
- </td>
61
- </tr>
62
- <tr>
63
- <td><p style="text-align: center">Car Window</p></td>
64
- <td><p style="text-align: center">Very Green Office</p></td>
65
- </tr>
66
- <tr>
67
- <td>
68
- <div style="width: 100%; max-width: 600px; position: relative;">
69
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/Input_car.jpg?raw=true" style="width: 100%; height: 300px; display: block;">
70
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/Ours_car.png?raw=true" style="width: 100%; height: 300px; display: block; position: absolute; top: 0; left: 0; opacity: 0; transition: opacity 0.5s;" onmouseover="this.style.opacity=1;" onmouseout="this.style.opacity=0;">
71
- </div>
72
- </td>
73
- <td>
74
- <div style="width: 100%; max-width: 600px; position: relative;">
75
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/Input_green.png?raw=true" style="width: 100%; height: 300px; display: block;">
76
- <img src="https://github.com/lime-j/RDNet/blob/main/figures/Ours_green.png?raw=true" style="width: 100%; height: 300px; display: block; position: absolute; top: 0; left: 0; opacity: 0; transition: opacity 0.5s;" onmouseover="this.style.opacity=1;" onmouseout="this.style.opacity=0;">
77
- </div>
78
- </td>
79
- </tr>
80
- </table>
81
-
82
- ## Requirements
83
- We recommend torch 2.x for our code, but it should works fine with most of the modern versions.
84
-
85
- ```
86
- pip install torch>=2.0 torchvision
87
- pip install einops ema-pytorch fsspec fvcore huggingface-hub matplotlib numpy opencv-python omegaconf pytorch-msssim scikit-image scikit-learn scipy tensorboard tensorboardx wandb timm
88
- ```
89
-
90
- # Testing
91
- The checkpoint for the main network is available at https://checkpoints.mingjia.li/rdnet.pth ; while the model for cls_model is at https://checkpoints.mingjia.li/cls_model.pth . Please put the cls_model.pth under "pretrained" folder.
92
-
93
- ```python
94
- python3 test_sirs.py --icnn_path <path to the main checkpoint> --resume
95
- ```
96
- # Training
97
-
98
- Training script will be released in a few days.
99
-
 
1
+ title: RDNet
2
+ emoji: 💻
3
+ colorFrom: indigo
4
+ colorTo: blue
5
+ sdk: gradio
6
+ sdk_version: 4.43.0
7
+ app_file: app.py
8
+ pinned: false