Skip to content

Latest commit

 

History

History
38 lines (31 loc) · 2.5 KB

INSTALL.md

File metadata and controls

38 lines (31 loc) · 2.5 KB

Installation

Set up the environment

conda create -n agenticir python=3.10
conda activate agenticir
pip install -r installation/requirements.txt

Deploy IR models

We employ the following models as single-degradation restoration tools: DiffBIR, X-Restormer, SwinIR, HAT, MPRNet, MAXIM, Restormer, DRBNet, IFAN, RIDCP, DehazeFormer, and FBCNN. Thanks for these awesome works.

  • Run sh installation/deploy_tools.sh to prepare the code, which is adapted from the official repos linked above.
  • Set up their respective environments according to the official repos.
  • Download the weights. You may need to modify the paths in files in executor. A tutorial will be given if necessary.
  • Run python -m test_tool.test_tool to check whether all tools work properly.

Deploy DepictQA

  • Run sh installation/deploy_depictqa.sh to prepare the code, which is adapted from the official repo linked above.

  • Set up the environment according to the official repo.

  • Download the weights.

    • Download the pre-trained ViT from this link and put it in DepictQA/weights/.
    • Download the pre-trained Vicuna from this link and put it in DepictQA/weights/.
    • Download the delta weights of DepictQA-Wild from this link, rename it to DQ495K.pt, and put it in DepictQA/weights/delta/.
    • Download the delta weights fine-tuned from DepictQA-Wild from this link and put it in DepictQA/weights/delta/.

    The structure of DepictQA/weights should look like this:

    DepictQA/weights/
    ├── ViT-L-14.pt
    ├── vicuna-7b-v1.5/
    │   └── ...
    └── delta/
        ├── DQ495K.pt
        └── degra_eval.pt