Update README.md
Browse files
README.md
CHANGED
|
@@ -14,7 +14,16 @@ tags:
|
|
| 14 |
|
| 15 |
HiT-SR is a general strategy to improve transformer-based SR methods. We apply our HiT-SR approach to improve [SwinIR-Light](https://github.com/JingyunLiang/SwinIR), [SwinIR-NG](https://github.com/rami0205/NGramSwin) and [SRFormer-Light](https://github.com/HVision-NKU/SRFormer), corresponding to our HiT-SIR, HiT-SNG, and HiT-SRF. Compared with the original structure, our improved models achieve better SR performance while reducing computational burdens.
|
| 16 |
|
| 17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
For each HiT-SR model, we provide 2x, 3x, 4x upscaling versions:
|
| 19 |
| Repo Name | | Model | | Upscale |
|
| 20 |
|-------------------|---|---------|---|---------|
|
|
@@ -28,17 +37,6 @@ For each HiT-SR model, we provide 2x, 3x, 4x upscaling versions:
|
|
| 28 |
| `XiangZ/hit-srf-3x` | | HiT-SRF | | 3x |
|
| 29 |
| `XiangZ/hit-srf-4x` | | HiT-SRF | | 4x |
|
| 30 |
|
| 31 |
-
|
| 32 |
-
## 🛠️ Setup
|
| 33 |
-
Install the dependencies under the working directory:
|
| 34 |
-
```
|
| 35 |
-
git clone https://huggingface.co/XiangZ/hit-sr
|
| 36 |
-
cd hit-sr
|
| 37 |
-
pip install -r requirements.txt
|
| 38 |
-
```
|
| 39 |
-
|
| 40 |
-
## 🚀 Usage
|
| 41 |
-
|
| 42 |
To test the model (use hit-srf-4x as an example):
|
| 43 |
```
|
| 44 |
from hit_sir_arch import HiT_SIR
|
|
|
|
| 14 |
|
| 15 |
HiT-SR is a general strategy to improve transformer-based SR methods. We apply our HiT-SR approach to improve [SwinIR-Light](https://github.com/JingyunLiang/SwinIR), [SwinIR-NG](https://github.com/rami0205/NGramSwin) and [SRFormer-Light](https://github.com/HVision-NKU/SRFormer), corresponding to our HiT-SIR, HiT-SNG, and HiT-SRF. Compared with the original structure, our improved models achieve better SR performance while reducing computational burdens.
|
| 16 |
|
| 17 |
+
|
| 18 |
+
## 🛠️ Setup
|
| 19 |
+
Install the dependencies under the working directory:
|
| 20 |
+
```
|
| 21 |
+
git clone https://huggingface.co/XiangZ/hit-sr
|
| 22 |
+
cd hit-sr
|
| 23 |
+
pip install -r requirements.txt
|
| 24 |
+
```
|
| 25 |
+
|
| 26 |
+
## 🚀 Usage
|
| 27 |
For each HiT-SR model, we provide 2x, 3x, 4x upscaling versions:
|
| 28 |
| Repo Name | | Model | | Upscale |
|
| 29 |
|-------------------|---|---------|---|---------|
|
|
|
|
| 37 |
| `XiangZ/hit-srf-3x` | | HiT-SRF | | 3x |
|
| 38 |
| `XiangZ/hit-srf-4x` | | HiT-SRF | | 4x |
|
| 39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
To test the model (use hit-srf-4x as an example):
|
| 41 |
```
|
| 42 |
from hit_sir_arch import HiT_SIR
|