单目深度估计【DINOv2】

深度学习界扛把子

已于 2023-05-24 17:36:57 修改

阅读量703

点赞数

文章标签：单目深度估计

于 2023-05-24 17:23:35 首次发布

本文链接：https://blog.csdn.net/qq_16792139/article/details/130851748

版权

https://github.com/facebookresearch/dinov2
https://github.com/facebookresearch/dinov2/blob/main/MODEL_CARD.md

The models can be used without fine-tuning, with downstream classifiers as simple as linear layers, to obtain competitive results:

on depth estimation, semantic segmentation, using linear layers.

里面介绍了深度估计用线性层
使用下面的代码开始使用该模型。

import torch
dinov2_vits14 = torch.hub.load('facebookresearch/dinov2', 'dinov2_vits14')
dinov2_vitb14 = torch.hub.load('facebookresearch/dinov2', 'dinov2_vitb14')
dinov2_vitl14 = torch.hub.load('facebookresearch/dinov2', 'dinov2_vitl14')
dinov2_vitg14 = torch.hub.load('facebookresearch/dinov2', 'dinov2_vitg14')