-
+
|
|
+:-------------------------:|:-------------------------:
+Cover figure | Network structure
](https://imgsli.com/Mjc5ODI4)
+
+### Outdoor-Rain
+[
](
+https://imgsli.com/Mjc5ODIy)
+
+
+## Installation
+
+See [INSTALL.md](INSTALL.md) for the installation of dependencies required to run Histoformer.
+
+## Training
+
+1. Download [Training set](https://drive.google.com/file/d/1tfeBnjZX1wIhIFPl6HOzzOKOyo0GdGHl/view?usp=sharing) or each of them, i.e., [Snow100K](https://sites.google.com/view/yunfuliu/desnownet), [Outdoor-Rain](https://github.com/liruoteng/HeavyRainRemoval), and [RainDrop](https://github.com/rui1996/DeRaindrop).
+
+2. To train Histoformer with default settings, run
+```
+cd Histoformer
+./train.sh Allweather/Options/Allweather_Histoformer.yml
+```
+
+**Note:** The above training script uses 4 GPUs by default.
+To use any other settings, modify [Histoformer/train.sh](../train.sh) and [Allweather/Options/Allweather_Histoformer.yml](Options/Allweather_Histoformer.yml)
+
+## Evaluation
+
+1. Download the pre-trained [models](https://drive.google.com/drive/folders/1dmPhr8Z5iPRx9lh7TwdUFPSfwGIxp5l0?usp=drive_link) and place it in `./pretrained_models/`
+
+2. Download test datasets from each of them, i.e., [Snow100K](https://sites.google.com/view/yunfuliu/desnownet), [Outdoor-Rain](https://github.com/liruoteng/HeavyRainRemoval), and [RainDrop](https://github.com/rui1996/DeRaindrop).
+
+3. Testing with the replaced argument ```--input_dir [INPUT_FOLDER]```
+```
+python test_histoformer.py --input_dir [INPUT_FOLDER] --result_dir result/ --weights pretrained_models/net_g_best.pth --yaml_file Options/Allweather_Histoformer.yml
+
+# for realsnow
+python test_histoformer.py --input_dir [INPUT_FOLDER] --result_dir result/ --weights pretrained_models/net_g_real.pth --yaml_file Options/Allweather_Histoformer.yml
+```
+
+
+
+## Citation
+If you use Restormer, please consider citing:
+
+ @misc{sun2024restoring,
+ title={Restoring Images in Adverse Weather Conditions via Histogram Transformer},
+ author={Shangquan Sun and Wenqi Ren and Xinwei Gao and Rui Wang and Xiaochun Cao},
+ year={2024},
+ eprint={2407.10172},
+ archivePrefix={arXiv},
+ primaryClass={cs.CV}
+}
+
+## Contact
+If you have any question, please contact shangquansun@gmail.com
+
+**Acknowledgment:** This code is based on the [Restormer](https://github.com/swz30/Restormer), [BasicSR](https://github.com/xinntao/BasicSR) toolbox and [HINet](https://github.com/megvii-model/HINet).
+
In progress!
\ No newline at end of file
diff --git a/assets/example-Outdoor-Rain.png b/assets/example-Outdoor-Rain.png
new file mode 100644
index 0000000000000000000000000000000000000000..9990570bdbe5d38ff2a0c3521b546f031a570cf2
GIT binary patch
literal 418128
zcmV*UKwH0wP)