Dense Prediction Transformers For Python

Vision Transformers for Dense Prediction

This repository contains code and models for our paper:

Vision Transformers for Dense Prediction
René Ranftl, Alexey Bochkovskiy, Vladlen Koltun

Changelog

• [March 2021] Initial release of inference code and models

Setup

1. Download the model weights and place them in the weights folder:

Monodepth:

Segmentation:

2. Set up dependencies:

   pip install -r requirements.txt

   The code was tested with Python 3.7, PyTorch 1.8.0, OpenCV 4.5.1, and timm 0.4.5

Usage

1. Place one or more input images in the folder input.

 

 

 

To finish reading, please visit source site