Transformers 文档

DINOv3

Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

该模型于 2025 年 8 月 13 日在 HF papers 上发布,并于 2025 年 8 月 14 日贡献给 Hugging Face Transformers。

FlashAttention SDPA

DINOv3

DINOv3 是一系列通用视觉基础模型,无需微调即可在广泛的应用场景中超越特定领域的先进模型。DINOv3 可生成高质量的密集特征,在各种视觉任务上实现卓越表现,显著超越了以往的自监督和弱监督基础模型。

您可以在 DINOv3 合集中找到所有原始的 DINOv3 检查点(checkpoints)。

点击右侧边栏中的 DINOv3 模型,查看更多关于如何将 DINOv3 应用于不同视觉任务的示例。

下面的示例演示了如何使用 PipelineAutoModel 类获取图像嵌入(embedding)。

流水线
自动模型
from transformers import pipeline


pipe = pipeline(
    task="image-feature-extraction",
    model="facebook/dinov3-vits16-pretrain-lvd1689m",
)

pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/pipeline-cat-chonk.jpeg")

量化通过以较低精度表示权重来减少大型模型的内存负担。有关更多可用量化后端,请参阅量化概述。

以下示例使用torchao仅将权重量化为int4。

# pip install torchao
import torch
from torchao.quantization import Int4WeightOnlyConfig

from transformers import AutoImageProcessor, AutoModel, TorchAoConfig
from transformers.image_utils import load_image


url = "http://images.cocodataset.org/val2017/000000039769.jpg"
image = load_image(url)

processor = AutoImageProcessor.from_pretrained("facebook/dinov3-vitsplus-pretrain-lvd1689m")

quant_type = Int4WeightOnlyConfig(group_size=128)
quantization_config = TorchAoConfig(quant_type=quant_type)

model = AutoModel.from_pretrained(
    "facebook/dinov3-vit7b16-pretrain-lvd1689m",
    device_map="auto",
    quantization_config=quantization_config
)

inputs = processor(images=image, return_tensors="pt").to(model.device)
with torch.inference_mode():
    outputs = model(**inputs)

pooled_output = outputs.pooler_output
print("Pooled output shape:", pooled_output.shape)

注意事项

  • 下面的示例显示如何将输出张量拆分为

    • 整个图像的一个嵌入,通常称为 CLS token,适用于分类和检索任务
    • 寄存器 tokens(register tokens) - 作为全局信息专用“内存槽”的可学习嵌入,它们能减少 patch tokens 中的高范数伪影,从而生成更清晰的注意力图,并在密集预测任务上表现更好。
    • 一组局部嵌入,输入图像的每个 16x16 patch 对应一个,适用于语义分割等密集任务
    import torch
    from transformers import AutoImageProcessor, AutoModel
    from transformers.image_utils import load_image
    
    url = "http://images.cocodataset.org/val2017/000000039769.jpg"
    image = load_image(url)
    print("Image size:", image.height, image.width)  # [480, 640]
    
    processor = AutoImageProcessor.from_pretrained("facebook/dinov3-vits16-pretrain-lvd1689m")
    model = AutoModel.from_pretrained("facebook/dinov3-vits16-pretrain-lvd1689m", device_map="auto")
    patch_size = model.config.patch_size
    print("Patch size:", patch_size) # 16
    print("Num register tokens:", model.config.num_register_tokens) # 4
    
    inputs = processor(images=image, return_tensors="pt").to(model.device)
    print("Preprocessed image size:", inputs.pixel_values.shape)  # [1, 3, 224, 224]
    
    batch_size, _, img_height, img_width = inputs.pixel_values.shape
    num_patches_height, num_patches_width = img_height // patch_size, img_width // patch_size
    num_patches_flat = num_patches_height * num_patches_width
    
    with torch.inference_mode():
      outputs = model(**inputs)
    
    last_hidden_states = outputs.last_hidden_state
    print(last_hidden_states.shape)  # [1, 1 + 4 + 256, 384]
    assert last_hidden_states.shape == (batch_size, 1 + model.config.num_register_tokens + num_patches_flat, model.config.hidden_size)
    
    cls_token = last_hidden_states[:, 0, :]
    patch_features_flat = last_hidden_states[:, 1 + model.config.num_register_tokens:, :]
    patch_features = patch_features_flat.unflatten(1, (num_patches_height, num_patches_width))

DINOv3ViTConfig

class transformers.DINOv3ViTConfig

< >

( transformers_version: str | None = None architectures: list[str] | None = None output_hidden_states: bool | None = False return_dict: bool | None = True dtype: typing.Union[str, ForwardRef('torch.dtype'), NoneType] = None chunk_size_feed_forward: int = 0 is_encoder_decoder: bool = False id2label: dict[int, str] | dict[str, str] | None = None label2id: dict[str, int] | dict[str, str] | None = None problem_type: typing.Optional[typing.Literal['regression', 'single_label_classification', 'multi_label_classification']] = None patch_size: int | list[int] | tuple[int, int] = 16 hidden_size: int = 384 intermediate_size: int = 1536 num_hidden_layers: int = 12 num_attention_heads: int = 6 hidden_act: str = 'gelu' attention_dropout: float | int = 0.0 initializer_range: float = 0.02 layer_norm_eps: float = 1e-05 rope_theta: float = 100.0 image_size: int | list[int] | tuple[int, int] = 224 num_channels: int = 3 query_bias: bool = True key_bias: bool = False value_bias: bool = True proj_bias: bool = True mlp_bias: bool = True layerscale_value: float = 1.0 drop_path_rate: float | int = 0.0 use_gated_mlp: bool = False num_register_tokens: int = 0 pos_embed_shift: float | None = None pos_embed_jitter: float | None = None pos_embed_rescale: float | None = 2.0 _out_features: list[str] | None = None _out_indices: list[int] | None = None apply_layernorm: bool = True reshape_hidden_states: bool = True )

参数

  • patch_size (Union[int, list[int], tuple[int, int]], 可选, 默认为 16) — 每个 patch 的大小(分辨率)。
  • hidden_size (int, 可选, 默认为 384) — 隐藏表示的维度。
  • intermediate_size (int, 可选, 默认为 1536) — MLP 表示的维度。
  • num_hidden_layers (int, 可选, 默认为 12) — Transformer 解码器中的隐藏层数量。
  • num_attention_heads (int, 可选, 默认为 6) — Transformer 解码器中每个注意力层的注意力头数量。
  • hidden_act (str, 可选, 默认为 gelu) — 解码器中的非线性激活函数(函数或字符串)。例如:"gelu", "relu", "silu" 等。
  • attention_dropout (Union[float, int], 可选, 默认为 0.0) — 注意力概率的 Dropout 比率。
  • initializer_range (float, 可选, 默认为 0.02) — 用于初始化所有权重矩阵的截断正态初始化器(truncated_normal_initializer)的标准差。
  • layer_norm_eps (float, 可选, 默认为 1e-05) — 层归一化层使用的 epsilon 值。
  • rope_theta (float, 可选, 默认为 100.0) — RoPE 嵌入的基周期。
  • image_size (Union[int, list[int], tuple[int, int]], 可选, 默认为 224) — 每个图像的大小(分辨率)。
  • num_channels (int, 可选, 默认为 3) — 输入通道的数量。
  • query_bias (bool, 可选, 默认为 True) — 是否在 query 投影中添加偏置(bias)。
  • key_bias (bool, 可选, 默认为 False) — 是否在 key 投影中添加偏置。
  • value_bias (bool, 可选, 默认为 True) — 是否在 value 投影中添加偏置。
  • proj_bias (bool, 可选, 默认为 True) — 是否在输出投影中添加偏置。
  • mlp_bias (bool, 可选, 默认为 True) — 是否在 MLP 层中的 up_proj、down_proj 和 gate_proj 层中使用偏置。
  • layerscale_value (float, 可选, 默认为 1.0) — 用于层缩放(layer scale)的初始值。
  • drop_path_rate (Union[float, int], 可选, 默认为 0.0) — 用于 patch 融合的 Drop path 比率。
  • use_gated_mlp (bool, 可选, 默认为 False) — 是否使用 SwiGLU 前馈神经网络。
  • num_register_tokens (int, 可选, 默认为 0) — 寄存器 token 的数量。
  • pos_embed_shift (float, 可选) — 随机偏移位置嵌入坐标的量(范围在 [-shift, shift] 内),如果不为 None,则仅在训练模式下应用。
  • pos_embed_jitter (float, 可选) — 随机抖动位置嵌入坐标的量(以对数均匀值在 [1/jitter, jitter] 范围内),如果不为 None,则仅在训练模式下应用。
  • pos_embed_rescale (float, 可选, 默认为 2.0) — 随机重缩放位置嵌入坐标的量(以对数均匀值在 [1/rescale, rescale] 范围内),如果不为 None,则仅在训练模式下应用。
  • apply_layernorm (bool, optional, 默认为 True) — 当作为骨干网络 (backbone) 使用时,是否对特征图应用层归一化 (layer normalization)。
  • reshape_hidden_states (bool, optional, 默认为 True) — 当作为骨干网络 (backbone) 使用时,是否将隐藏状态重塑为空间维度。

这是用于存储 DINOv3ViTModel 配置的配置类。它用于根据指定的参数实例化一个 Dinov3 Vit 模型,从而定义模型架构。使用默认值实例化配置将生成与 facebook/dinov3-vits16-pretrain-lvd1689m 类似的配置。

配置对象继承自 PreTrainedConfig,可用于控制模型输出。阅读 PreTrainedConfig 的文档以获取更多信息。

示例

>>> from transformers import DINOv3ViTConfig, DINOv3ViTModel

>>> # Initializing a DINOv3 ViT-small style configuration
>>> config = DINOv3ViTConfig()

>>> # Initializing a model (with random weights) from the config
>>> model = DINOv3ViTModel(config)

>>> # Accessing the model config
>>> config = model.config

DINOv3ConvNextConfig

class transformers.DINOv3ConvNextConfig

< >

( transformers_version: str | None = None architectures: list[str] | None = None output_hidden_states: bool | None = False return_dict: bool | None = True dtype: typing.Union[str, ForwardRef('torch.dtype'), NoneType] = None chunk_size_feed_forward: int = 0 is_encoder_decoder: bool = False id2label: dict[int, str] | dict[str, str] | None = None label2id: dict[str, int] | dict[str, str] | None = None problem_type: typing.Optional[typing.Literal['regression', 'single_label_classification', 'multi_label_classification']] = None num_channels: int = 3 hidden_sizes: list[int] | None = None depths: list[int] | None = None hidden_act: str = 'gelu' initializer_range: float = 0.02 layer_norm_eps: float = 1e-06 layer_scale_init_value: float = 1e-06 drop_path_rate: float | int = 0.0 image_size: int | list[int] | tuple[int, int] = 224 _out_features: list[str] | None = None _out_indices: list[int] | None = None )

参数

  • num_channels (int, optional, 默认为 3) — 输入通道数。
  • hidden_sizes (list[int], optional) — 模型每一阶段的维度(隐藏层大小)。
  • depths (list[int], optional) — Transformer 中每一层的深度。
  • hidden_act (str, optional, 默认为 gelu) — 解码器中的非线性激活函数(函数或字符串)。例如:"gelu", "relu", "silu" 等。
  • initializer_range (float, optional, 默认为 0.02) — 用于初始化所有权重矩阵的 truncated_normal_initializer 的标准差。
  • layer_norm_eps (float, optional, 默认为 1e-06) — 层归一化层使用的 epsilon 值。
  • layer_scale_init_value (float, optional, 默认为 1e-06) — 自注意力层中使用的缩放比例。Base 模型为 0.1,Large 模型为 1e-6。设为 0 可禁用层缩放。
  • drop_path_rate (Union[float, int], optional, 默认为 0.0) — 用于 Patch 融合 (patch fusion) 的路径丢弃率 (drop path rate)。
  • image_size (Union[int, list[int], tuple[int, int]], optional, 默认为 224) — 每张图像的大小(分辨率)。

这是用于存储 DINOv3ConvNextModel 配置的配置类。它用于根据指定的参数实例化一个 Dinov3 Convnext 模型,从而定义模型架构。使用默认值实例化配置将生成与 facebook/dinov3-convnext-tiny-pretrain-lvd1689m 类似的配置。

配置对象继承自 PreTrainedConfig,可用于控制模型输出。阅读 PreTrainedConfig 的文档以获取更多信息。

示例

>>> from transformers import DINOv3ConvNextConfig, DINOv3ConvNextModel

>>> # Initializing a DINOv3ConvNext (tiny variant) style configuration
>>> config = DINOv3ConvNextConfig()

>>> # Initializing a model (with random weights)
>>> model = DINOv3ConvNextModel(config)

>>> # Accessing the model config
>>> config = model.config

DINOv3ViTModel

class transformers.DINOv3ViTModel

< >

( config: DINOv3ViTConfig )

参数

  • config (DINOv3ViTConfig) — 模型配置类,包含模型的所有参数。使用配置文件初始化不会加载模型相关的权重,仅加载配置。查看 from_pretrained() 方法以加载模型权重。

Dinov3 Vit 基础模型,输出原始隐藏状态,顶部没有任何特定的分类头。

该模型继承自 PreTrainedModel。请查看超类文档以了解该库为所有模型实现的通用方法(例如下载或保存、调整输入嵌入大小、剪枝头部等)。

此模型也是一个 PyTorch torch.nn.Module 子类。像普通的 PyTorch Module 一样使用它,并参考 PyTorch 文档了解一般用法和行为的所有相关信息。

forward

< >

( pixel_values: Tensor bool_masked_pos: torch.Tensor | None = None **kwargs: typing_extensions.Unpack[transformers.utils.generic.TransformersKwargs] ) BaseModelOutputWithPoolingtuple(torch.FloatTensor)

参数

  • pixel_values (形状为 (batch_size, num_channels, image_size, image_size)torch.Tensor) — 对应于输入图像的张量。像素值可以使用 DINOv3ViTImageProcessor 获取。详情请参见 DINOv3ViTImageProcessor.__call__()processor_class 使用 DINOv3ViTImageProcessor 来处理图像)。
  • bool_masked_pos (形状为 (batch_size, sequence_length)torch.BoolTensor) — 布尔遮蔽位置。指示哪些 patch 被遮蔽 (1),哪些没有被遮蔽 (0)。仅在预训练时相关。

返回

BaseModelOutputWithPooling or tuple(torch.FloatTensor)

BaseModelOutputWithPooling 或一个 torch.FloatTensor 元组(如果传入了 return_dict=False 或当 config.return_dict=False 时),根据配置(DINOv3ViTConfig)和输入包含各种元素。

DINOv3ViTModel 的 forward 方法,重写了 __call__ 特殊方法。

虽然 forward pass 的实现需要在此函数中定义,但你应该在之后调用 Module 实例而不是这个,因为前者负责运行预处理和后处理步骤,而后者会静默地忽略它们。

  • last_hidden_state (torch.FloatTensor, 形状为 (batch_size, sequence_length, hidden_size)) — 模型最后一层输出的隐藏状态序列。

  • pooler_output (torch.FloatTensor,形状为 (batch_size, hidden_size)) — 序列第一个 token(分类 token)在进一步通过用于辅助预训练任务的层后的最后一个隐藏状态。例如,对于 BERT 系列模型,这会返回经过线性层和 tanh 激活函数处理后的分类 token。线性层的权重是通过预训练期间的下一句预测(分类)目标来训练的。

  • hidden_states (tuple(torch.FloatTensor), optional, 当传递 output_hidden_states=True 或当 config.output_hidden_states=True 时返回) — torch.FloatTensor 的元组(一个用于嵌入层的输出,如果模型有嵌入层;+一个用于每个层的输出),形状为 (batch_size, sequence_length, hidden_size)

    模型在每个层输出的隐藏状态以及可选的初始嵌入输出。

  • attentions (tuple(torch.FloatTensor), optional, 当传递 output_attentions=True 或当 config.output_attentions=True 时返回) — torch.FloatTensor 的元组(每个层一个),形状为 (batch_size, num_heads, sequence_length, sequence_length)

    注意力 softmax 后的注意力权重,用于计算自注意力头中的加权平均值。

示例

DINOv3ViTBackbone

class transformers.DINOv3ViTBackbone

< >

( config model_args: ~utils.generic.ModelArgs | None = None adapter_args: ~utils.generic.AdapterArgs | None = None lora_args: ~utils.generic.LoRAArgs | None = None tokenizer_args: ~utils.generic.TokenizerArgs | None = None dataset_args: ~utils.generic.DatasetArgs | None = None data_args: ~utils.generic.DataArgs | None = None training_args: ~utils.generic.TrainingArgs | None = None generation_args: ~utils.generic.GenerationArgs | None = None vision_tower_args: ~utils.generic.VisionTowerArgs | None = None qlora_args: ~utils.generic.QLoRAArgs | None = None vision_tower_template_args: ~utils.generic.VisionTowerTemplateArgs | None = None video_tower_args: ~utils.generic.VideoTowerArgs | None = None vision_config: ~utils.generic.VisionConfig | None = None video_config: ~utils.generic.VideoConfig | None = None load_dataset: bool | None = None load_data_collator: bool | None = None load_processor: bool | None = None load_lora_adapter: bool | None = None load_adapter: bool | None = None load_qlora_adapter: bool | None = None **kwargs: typing_extensions.Unpack[transformers.modeling_utils.PreTrainedModelKwargs] )

参数

  • config (DINOv3ViTBackbone) — 模型配置类,包含模型的所有参数。使用配置文件初始化不会加载模型相关的权重,仅加载配置。查看 from_pretrained() 方法以加载模型权重。

Dinov3 Vit 骨干网络。

该模型继承自 PreTrainedModel。请查看超类文档以了解该库为所有模型实现的通用方法(例如下载或保存、调整输入嵌入大小、剪枝头部等)。

此模型也是一个 PyTorch torch.nn.Module 子类。像普通的 PyTorch Module 一样使用它,并参考 PyTorch 文档了解一般用法和行为的所有相关信息。

forward

< >

( pixel_values: Tensor **kwargs: typing_extensions.Unpack[transformers.utils.generic.TransformersKwargs] ) DINOv3ViTBackboneOutputtuple(torch.FloatTensor)

参数

  • pixel_values (形状为 (batch_size, num_channels, image_size, image_size)torch.Tensor) — 对应于输入图像的张量。像素值可以使用 DINOv3ViTImageProcessor 获取。详情请参见 DINOv3ViTImageProcessor.__call__()processor_class 使用 DINOv3ViTImageProcessor 来处理图像)。

返回

DINOv3ViTBackboneOutputtuple(torch.FloatTensor)

一个 DINOv3ViTBackboneOutput 或一个 torch.FloatTensor 元组(如果传入了 return_dict=False 或当 config.return_dict=False 时),根据配置(DINOv3ViTConfig)和输入包含各种元素。

DINOv3ViTBackbone 的 forward 方法,重写了 __call__ 特殊方法。

虽然 forward pass 的实现需要在此函数中定义,但你应该在之后调用 Module 实例而不是这个,因为前者负责运行预处理和后处理步骤,而后者会静默地忽略它们。

  • feature_maps (tuple(torch.FloatTensor) of shape (batch_size, num_channels, height, width)) — 阶段的特征图。

  • hidden_states (tuple(torch.FloatTensor), optional, returned when output_hidden_states=True is passed or when config.output_hidden_states=True) — torch.FloatTensor 元组(一个用于嵌入输出 + 每个层输出一个)的形状为 (batch_size, sequence_length, hidden_size)(batch_size, num_channels, height, width),具体取决于主干网络。

    模型在每个阶段输出的隐藏状态以及初始嵌入输出。

  • attentions (tuple(torch.FloatTensor), optional, returned when output_attentions=True is passed or when config.output_attentions=True) — torch.FloatTensor 元组(每个层一个)的形状为 (batch_size, num_heads, sequence_length, sequence_length)。仅当主干网络使用注意力时适用。

    注意力 softmax 后的注意力权重,用于计算自注意力头中的加权平均值。

  • cls_tokens (tuple(torch.FloatTensor), optional) — 来自每个选定特征阶段的 CLS token,每个形状为 (batch_size, hidden_size)。仅在 config.return_class_token=True 时存在。

DINOv3ConvNextModel

class transformers.DINOv3ConvNextModel

< >

( config: DINOv3ConvNextConfig )

参数

  • config (DINOv3ConvNextConfig) — 包含模型所有参数的模型配置类。使用配置文件初始化模型不会加载相关的权重,只会加载配置。请查看 from_pretrained() 方法以加载模型权重。

原始的 Dinov3 Convnext 模型,输出原始隐藏状态,顶部没有任何特定的预测头。

该模型继承自 PreTrainedModel。请查看超类文档以了解该库为所有模型实现的通用方法(例如下载或保存、调整输入嵌入大小、剪枝头部等)。

此模型也是一个 PyTorch torch.nn.Module 子类。像普通的 PyTorch Module 一样使用它,并参考 PyTorch 文档了解一般用法和行为的所有相关信息。

forward

< >

( pixel_values: FloatTensor **kwargs ) BaseModelOutputWithPoolingAndNoAttentiontuple(torch.FloatTensor)

参数

  • pixel_values (形状为 (batch_size, num_channels, image_size, image_size)torch.FloatTensor) — 输入图像对应的张量。像素值可以使用 image_processor_class 获取。有关详细信息,请参阅 image_processor_class.__call__processor_class 使用 image_processor_class 来处理图像)。

返回

BaseModelOutputWithPoolingAndNoAttentiontuple(torch.FloatTensor)

一个 BaseModelOutputWithPoolingAndNoAttention 或一个 torch.FloatTensor 元组(如果传入了 return_dict=Falseconfig.return_dict=False 时),根据配置(DINOv3ConvNextConfig)和输入,包含各种不同的元素。

DINOv3ConvNextModel 的 forward 方法,重写了 __call__ 特殊方法。

虽然 forward pass 的实现需要在此函数中定义,但你应该在之后调用 Module 实例而不是这个,因为前者负责运行预处理和后处理步骤,而后者会静默地忽略它们。

  • last_hidden_state (torch.FloatTensor, 形状为 (batch_size, num_channels, height, width)) — 模型最后一层输出的隐藏状态序列。

  • pooler_output (torch.FloatTensor, 形状为 (batch_size, hidden_size)) — 经过空间维度池化操作后的最后一层隐藏状态。

  • hidden_states (tuple(torch.FloatTensor), optional, 当传入 output_hidden_states=Trueconfig.output_hidden_states=True 时返回) — torch.FloatTensor 的元组(如果模型有嵌入层,则包含一个嵌入层输出,加上每层的一个输出),形状为 (batch_size, num_channels, height, width)

    模型在每个层输出的隐藏状态以及可选的初始嵌入输出。

示例

DINOv3ViTImageProcessor

class transformers.DINOv3ViTImageProcessor

< >

( **kwargs: typing_extensions.Unpack[transformers.processing_utils.ImagesKwargs] )

参数

  • ***kwargs** (ImagesKwargs, 可选) — 额外的图像预处理选项。特定于模型的 kwargs 列在上方;有关支持参数的完整列表,请参阅 TypedDict 类。

构造一个 DINOv3ViTImageProcessor 图像处理器。

preprocess

< >

( images: typing.Union[ForwardRef('PIL.Image.Image'), numpy.ndarray, ForwardRef('torch.Tensor'), list['PIL.Image.Image'], list[numpy.ndarray], list['torch.Tensor']] *args **kwargs: typing_extensions.Unpack[transformers.processing_utils.ImagesKwargs] ) ~image_processing_base.BatchFeature

参数

  • images (Union[PIL.Image.Image, numpy.ndarray, torch.Tensor, list[PIL.Image.Image], list[numpy.ndarray], list[torch.Tensor]]) — 要预处理的图像。期望输入单个图像或一批像素值在 0 到 255 之间的图像。如果传入像素值在 0 到 1 之间的图像,请设置 do_rescale=False
  • return_tensors (strTensorType, 可选) — 如果设置为 'pt',则返回堆叠后的张量,否则返回张量列表。
  • ***kwargs** (ImagesKwargs, 可选) — 额外的图像预处理选项。特定于模型的 kwargs 列在上方;有关支持参数的完整列表,请参阅 TypedDict 类。

返回

~image_processing_base.BatchFeature

  • data (dict) — 由 call 方法返回的列表/数组/张量字典(“pixel_values”等)。
  • tensor_type (Union[None, str, TensorType], optional) — 您可以在此处提供 tensor_type 以在初始化时将整数列表转换为 PyTorch/Numpy 张量。

DINOv3ConvNextBackbone

class transformers.DINOv3ConvNextBackbone

< >

( config: DINOv3ConvNextConfig )

参数

  • config (DINOv3ConvNextConfig) — 包含模型所有参数的模型配置类。使用配置文件初始化模型不会加载相关的权重,只会加载配置。请查看 from_pretrained() 方法以加载模型权重。

Dinov3 Convnext 骨干网络。

该模型继承自 PreTrainedModel。请查看超类文档以了解该库为所有模型实现的通用方法(例如下载或保存、调整输入嵌入大小、剪枝头部等)。

此模型也是一个 PyTorch torch.nn.Module 子类。像普通的 PyTorch Module 一样使用它,并参考 PyTorch 文档了解一般用法和行为的所有相关信息。

forward

< >

( pixel_values: FloatTensor **kwargs ) BackboneOutputtuple(torch.FloatTensor)

参数

  • pixel_values (形状为 (batch_size, num_channels, image_size, image_size)torch.FloatTensor) — 输入图像对应的张量。像素值可以使用 image_processor_class 获取。有关详细信息,请参阅 image_processor_class.__call__processor_class 使用 image_processor_class 来处理图像)。

返回

BackboneOutputtuple(torch.FloatTensor)

一个 BackboneOutput 或一个 torch.FloatTensor 元组(如果传入了 return_dict=Falseconfig.return_dict=False 时),根据配置(DINOv3ConvNextConfig)和输入,包含各种不同的元素。

DINOv3ConvNextBackbone 的 forward 方法,重写了 __call__ 特殊方法。

虽然 forward pass 的实现需要在此函数中定义,但你应该在之后调用 Module 实例而不是这个,因为前者负责运行预处理和后处理步骤,而后者会静默地忽略它们。

  • feature_maps (tuple(torch.FloatTensor) of shape (batch_size, num_channels, height, width)) — 阶段的特征图。

  • hidden_states (tuple(torch.FloatTensor), optional, returned when output_hidden_states=True is passed or when config.output_hidden_states=True) — torch.FloatTensor 元组(一个用于嵌入输出 + 每个层输出一个)的形状为 (batch_size, sequence_length, hidden_size)(batch_size, num_channels, height, width),具体取决于主干网络。

    模型在每个阶段输出的隐藏状态以及初始嵌入输出。

  • attentions (tuple(torch.FloatTensor), optional, returned when output_attentions=True is passed or when config.output_attentions=True) — torch.FloatTensor 元组(每个层一个)的形状为 (batch_size, num_heads, sequence_length, sequence_length)。仅当主干网络使用注意力时适用。

    注意力 softmax 后的注意力权重,用于计算自注意力头中的加权平均值。

在 GitHub 上更新

© . This site is unofficial and not affiliated with Hugging Face, Inc.