davit

(β˜… 377)

[ECCV 2022]Code for paper "DaViT: Dual Attention Vision Transformer"

  • .gitignore
  • inference.py
  • Introduction.md
  • LICENSE
  • README.md
  • train.py
  • validate.py

# Installation Guide

1. Get the code
git clone https://github.com/dingmyu/davit

Downloads the entire project code from GitHub to your computer.

cd davit

Moves into the project folder you just downloaded.

2. Official Install Script

Easy Recommended
Prerequisites
  • Python 3 Python is required to use pip.
pip install torch===1.9.0+cu102 torchvision===0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

Installs the package published on PyPI directly β€” no need to clone the source.

pip install thop pyyaml fvcore pillow==8.3.2

Installs the package published on PyPI directly β€” no need to clone the source.

pip install mmcv-full==1.3.0 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html

Installs the package published on PyPI directly β€” no need to clone the source.

pip install -v -e . # or "python setup.py develop"

Installs the package published on PyPI directly β€” no need to clone the source.

βœ… After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.

Pulled directly from this repo's README.

3. Docker

Easy
Prerequisites
  • Git Needed to download the project code from GitHub.
  • Docker Desktop Needed to build and run containers. Install it and keep it running in the background.
⚠️ This is a large repository, so this method may point to an internal sub-package rather than the actual core product. Check the full README as well.
docker build -f mmdet/docker/Dockerfile -t davit .

Builds a runnable image based on the Dockerfile.

docker run -p 8080:80 davit

Runs the built image as an actual container.

βœ… Run docker compose ps to check the containers are Up. If the README mentions a port, open http://localhost:PORT in your browser.

4. Python

Easy
Prerequisites
  • Git Needed to download the project code from GitHub.
  • Python 3 On Windows, be sure to check 'Add Python to PATH' during installation.
pip install torch===1.9.0+cu102 torchvision===0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

Installs the package published on PyPI directly β€” no need to clone the source.

pip install thop pyyaml fvcore pillow==8.3.2

Installs the package published on PyPI directly β€” no need to clone the source.

pip install mmcv-full==1.3.0 -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html

Installs the package published on PyPI directly β€” no need to clone the source.

pip install -r requirements/build.txt

Installs the Python libraries listed in requirements.txt (or similar).

pip install -v -e . # or "python setup.py develop"

Installs the package published on PyPI directly β€” no need to clone the source.

βœ… If it runs without errors and prints output in the terminal, it worked.

Pulled directly from this repo's README.

// repository documentation