dinov2 实例模型报错解决

AI算法网奇

已于 2025-02-18 21:36:29 修改

阅读量1.1k

点赞数 5

CC 4.0 BY-SA版权

分类专栏：深度学习宝典文章标签：深度学习

于 2023-12-15 16:49:43 首次发布

本文链接：https://blog.csdn.net/jacke121/article/details/135021016

深度学习宝典专栏收录该内容

1039 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文将介绍深度学习框架dinov2的使用，通过具体实例展示如何搭建模型、训练与优化，帮助读者理解并掌握该框架的核心功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2025实例

有时报错Remote end closed connection without response：

自己下载dinov2_vitl14_reg4_pretrain.pth：

2025实例

import torch
from torch import nn

class Dinov2Backbone(nn.Module):
    def __init__(self, name='dinov2_vitb14', pretrained=False, *args, **kwargs):
        super().__init__()
        self.name = name
        self.encoder = torch.hub.load('facebookresearch/dinov2', self.name, pretrained=pretrained)
        self.patch_size = self.encoder.patch_size
        self.embed_dim = self.encoder.embed_dim

    def forward(self, x):
        """
        Encode a RGB image using a ViT-backbone
        Args:
            - x: torch.Tensor of shape [bs,3,w,h]
        Return:

了解本专栏