chenjinrong commited on
Commit
ac457da
·
verified ·
1 Parent(s): c3d2972

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +63 -1
README.md CHANGED
@@ -1,3 +1,65 @@
1
  ---
2
- license: mit
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ license: apache-2.0
3
+ tags:
4
+ - image-dehazing
5
+ - image-restoration
6
+ - vision-transformer
7
+ - transformer
8
+ - low-level-vision
9
+ - pytorch
10
+ pipeline_tag: image-to-image
11
  ---
12
+
13
+ # ED-Former: Efficient Dehazing Transformer with Attention-Adaptive Feed-Forward Network
14
+
15
+ <div align="center">
16
+
17
+ [![Paper](https://img.shields.io/badge/Paper-ScienceDirect-blue.svg)](https://www.sciencedirect.com/science/article/abs/pii/S0923596526001578)
18
+ [![GitHub](https://img.shields.io/badge/GitHub-Repository-black?logo=github)](https://github.com/2697166190a-beep/ED-Former)
19
+ [![License](https://img.shields.io/badge/License-Apache%202.0-green.svg)](https://opensource.org/licenses/Apache-2.0)
20
+
21
+ </div>
22
+
23
+ Official model weights and implementation details for **ED-Former**, published in ***Signal Processing: Image Communication***.
24
+
25
+ ---
26
+
27
+ ## 🌟 Highlights & Model Specs
28
+
29
+ - **Attention-Adaptive Feed-Forward Network (AA-FFN)**: Dynamically refines feature representation while drastically cutting down computational overhead.
30
+ - **Ultra Lightweight & Fast**: Only **0.866 M** parameters and **7.36 G MACs** (evaluated on 256×256 input size).
31
+ - **State-of-the-Art Performance**: Achieves **38.21 dB PSNR** on SOTS-Indoor and **39.61 dB PSNR** on RS-Haze.
32
+
33
+ ---
34
+
35
+ ## 📊 Benchmark Results
36
+
37
+ Quantitative evaluation across multiple standard synthetic and real-world dehazing benchmarks:
38
+
39
+ | Benchmark Dataset | PSNR (dB) ↑ | SSIM ↑ | Model Weights |
40
+ | :--- | :---: | :---: | :---: |
41
+ | **SOTS-indoor (RESIDE-IN)** | **38.21** | **0.9942** | [Download](./) |
42
+ | **SOTS-outdoor (RESIDE-OUT)** | **33.92** | **0.9827** | [Download](./) |
43
+ | **RS-Haze** | **39.61** | **0.9715** | [Download](./) |
44
+ | **O-HAZE (zero-shot)** | **15.78** | **0.7020** | [Download](./) |
45
+
46
+ ### ⚡ Computational Efficiency:
47
+ - **Parameters (#Params)**: `0.866 M`
48
+ - **Computational Cost (#MACs)**: `7.36 G` (for $256 \times 256$ input)
49
+
50
+ ---
51
+
52
+ ## 🚀 Quick Start & Inference
53
+
54
+ Full training, testing scripts, and dataset setups are available at our **[GitHub Repository](https://github.com/2697166190a-beep/ED-Former)**.
55
+
56
+ ```bash
57
+ # 1. Clone the repository
58
+ git clone https://github.com/2697166190a-beep/ED-Former.git
59
+ cd ED-Former
60
+
61
+ # 2. Install dependencies
62
+ pip install -r requirements.txt
63
+
64
+ # 3. Run single image dehazing inference
65
+ python test.py --input_dir ./demo/hazy_images/ --weights ./weights/ed_former_sots.pth --output_dir ./results/