This repository is a refinement of the DPT network on the Season dataset.
Vision Transformers for Dense Prediction
René Ranftl, Alexey Bochkovskiy, Vladlen Koltun
For the official repository of DPT, see code.
It's an open-source monocular challenge that runs through February-May 2022. Please refer to the website for more detailed information.
The code was tested with Python 3.7, PyTorch 1.8.0, OpenCV 4.5.1, and timm 0.4.5
- We added other datasets to the official DPT pre-training weights to continue training, and finally trained 60 epochs in the Season Dataset.
DPT weight on Season Dataset in here.
python test.py --model_path --test_img_path
DPT(MIT License): https://github.com/isl-org/DPT#license