From 733871f8fbd935ea5b27e5e55aa3f9efe7b405a2 Mon Sep 17 00:00:00 2001 From: pp00704831 <57279608+pp00704831@users.noreply.github.com> Date: Sat, 16 Jul 2022 16:08:57 +0800 Subject: [PATCH] Update README.md --- README.md | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/README.md b/README.md index a63eb55..2d1b22e 100644 --- a/README.md +++ b/README.md @@ -2,3 +2,15 @@ Pytorch Implementation of "[Stripformer: Strip Transformer for Fast Image Deblurring](https://arxiv.org/abs/2204.04627)" + +## Installation +The implementation of our BANet is modified from "[DeblurGANv2](https://github.com/VITA-Group/DeblurGANv2)" +``` +git clone https://github.com/pp00704831/Stripformer.git +cd Stripformer +conda create -n Stripformer python=3.6 +source activate Stripformer +conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge +pip install opencv-python tqdm pyyaml joblib glog scikit-image tensorboardX albumentations==1.1.0 +pip install -U albumentations[imgaug] +```