This repository is built in PyTorch 1.8.1 and tested on Ubuntu 16.04 environment (Python3.7, CUDA10.2, cuDNN7.6). Follow these intructions
- Clone our repository
git clone https://github.com/sunshangquan/Histoformer.git
cd Histoformer
- Make conda environment
conda create -n pytorch181 python=3.7
conda activate pytorch181
- Install dependencies
conda install pytorch=1.8 torchvision cudatoolkit=10.2 -c pytorch
pip install matplotlib scikit-learn scikit-image opencv-python yacs joblib natsort h5py tqdm
pip install einops gdown addict future lmdb numpy pyyaml requests scipy tb-nightly yapf lpips
- Install basicsr
python setup.py develop --no_cuda_ext
To be able to download datasets automatically you would need go
and gdrive
installed.
- You can install
go
with the following
curl -O https://storage.googleapis.com/golang/go1.11.1.linux-amd64.tar.gz
mkdir -p ~/installed
tar -C ~/installed -xzf go1.11.1.linux-amd64.tar.gz
mkdir -p ~/go
- Add the lines in
~/.bashrc
export GOPATH=$HOME/go
export PATH=$PATH:$HOME/go/bin:$HOME/installed/go/bin
- Install
gdrive
using
go get github.com/prasmussen/gdrive
- Close current terminal and open a new terminal.