| # Low Light Rainy Code Export |
|
|
| This folder is a portable code-only export of `Low_light_rainy_new`. |
|
|
| ## Excluded |
|
|
| The following large/runtime folders were intentionally excluded: |
|
|
| - `dataset/` |
| - `checkpoint*/` |
| - `results/` |
| - Python caches |
| - `.pth`, `.zip`, `.log` files |
|
|
| ## Included Core Code |
|
|
| Key files for the current restoration experiments: |
|
|
| - `train_restormer.py` - main trainer for baseline, v1, and v2. |
| - `restormer.py` - original Restormer backbone and existing variants. |
| - `net/restormer_lowlight_rain.py` - FGP-Restormer v1. |
| - `net/restormer_lowlight_rain_v2.py` - FGP-Restormer v2. |
| - `utils/inference_utils.py` - validation padding, crop, RGB PSNR, optional TTA. |
| - `run_restormer_fgp.sh` - v1 launch script. |
| - `run_restormer_fgp_v2.sh` - v2 launch script. |
| - `dataset_RGB_ori.py`, `data_RGB.py` - legacy data loaders. |
| - `dataset_rwjr10k.py`, `train_restormer_rwjr10k.py`, `train_restormer_rwjr10k_from_original.py` - RWJR-10K related code, if needed. |
|
|
| ## Environment |
|
|
| The current `new` conda environment was exported to: |
|
|
| - `requirements_new.txt` |
| - `environment_new.yml` |
|
|
| Recommended setup on a new server: |
|
|
| ```bash |
| conda env create -f environment_new.yml |
| conda activate new |
| ``` |
|
|
| If the environment name conflicts, edit the first line of `environment_new.yml`, or use pip inside your own environment: |
|
|
| ```bash |
| python -m pip install -r requirements_new.txt |
| ``` |
|
|
| ## Data Layout Expected by Current v1/v2 Scripts |
|
|
| Default low-light-rain training expects: |
|
|
| ```text |
| dataset/train/syn+real/input |
| dataset/train/syn+real/target |
| dataset/test/input |
| dataset/test/target |
| ``` |
|
|
| The v1/v2 scripts do not include dataset files in this export. Copy datasets separately to the same relative locations or pass explicit paths: |
|
|
| ```bash |
| python train_restormer.py \ |
| --train_inp /path/to/input \ |
| --train_tar /path/to/target \ |
| --test_inp /path/to/test/input \ |
| --test_tar /path/to/test/target |
| ``` |
|
|
| ## Run v1 |
|
|
| ```bash |
| bash run_restormer_fgp.sh |
| ``` |
|
|
| ## Run v2 |
|
|
| ```bash |
| bash run_restormer_fgp_v2.sh |
| ``` |
|
|
| Before running on a new server, update any `--resume` paths in the run scripts if checkpoints are not present. |
|
|