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] +```