参数

• outputs (KeypointMatchingOutput) — 模型的原始输出。
• target_sizes (torch.Tensor 或 list[tuple[tuple[int, int]]], 可选) — 形状为 (batch_size, 2, 2) 的张量或元组列表（tuple[int, int]），包含批处理中每幅图像的目标尺寸 (height, width)。这必须是原始图像尺寸（在任何处理之前）。
• threshold (float, 可选, 默认为 0.0) — 过滤低分匹配的阈值。

字典列表，每个字典包含图像对中第一张和第二张图像的关键点、匹配分数和匹配索引。

参数

• images (ImageInput) — 要预处理的图像对。期望是包含 2 张图像的列表或包含 2 张图像列表的列表，像素值范围为 0 到 255。如果传入的图像像素值在 0 到 1 之间，请将 do_rescale=False。
• do_resize (bool, 可选, 默认为 self.do_resize) — 是否调整图像大小。
• size (dict[str, int], 可选, 默认为 self.size) — 应用 resize 后输出图像的尺寸。如果 size["shortest_edge"] >= 384，则图像被调整为 (size["shortest_edge"], size["shortest_edge"])。否则，图像的较短边将被匹配到 int(size["shortest_edge"]/ crop_pct)，之后图像被裁剪为 (size["shortest_edge"], size["shortest_edge"])。仅当 do_resize 设置为 True 时有效。
• resample (PILImageResampling, 可选, 默认为 self.resample) — 如果调整图像大小，则使用重采样过滤器。这可以是 PILImageResampling 的过滤器之一。仅当 do_resize 设置为 True 时有效。
• do_rescale (bool, 可选, 默认为 self.do_rescale) — 是否将图像值缩放到 [0 - 1] 之间。
• rescale_factor (float, 可选, 默认为 self.rescale_factor) — 如果 do_rescale 设置为 True，则按此缩放因子对图像进行缩放。
• do_grayscale (bool, 可选, 默认为 self.do_grayscale) — 是否将图像转换为灰度。
• return_tensors (str 或 TensorType, 可选) — 要返回的张量类型。可以是以下之一：
  • 未设置：返回 np.ndarray 列表。
  • TensorType.TENSORFLOW 或 'tf'：返回 tf.Tensor 类型的批处理。
  • TensorType.PYTORCH 或 'pt'：返回 torch.Tensor 类型的批处理。
  • TensorType.NUMPY 或 'np'：返回 np.ndarray 类型的批处理。
  • TensorType.JAX 或 'jax'：返回 jax.numpy.ndarray 类型的批处理。
• data_format (ChannelDimension 或 str, 可选, 默认为 ChannelDimension.FIRST) — 输出图像的通道维度格式。可以是以下之一：
  • "channels_first" 或 ChannelDimension.FIRST：图像格式为 (num_channels, height, width)。
  • "channels_last" 或 ChannelDimension.LAST：图像格式为 (height, width, num_channels)。
  • 未设置：使用输入图像的通道维度格式。
• input_data_format (ChannelDimension 或 str, 可选) — 输入图像的通道维度格式。如果未设置，则从输入图像推断通道维度格式。可以是以下之一：
  • "channels_first" 或 ChannelDimension.FIRST：图像格式为 (num_channels, height, width)。
  • "channels_last" 或 ChannelDimension.LAST：图像格式为 (height, width, num_channels)。
  • "none" 或 ChannelDimension.NONE：图像格式为 (height, width)。

参数

• image (np.ndarray) — 要调整大小的图像。
• size (dict[str, int]) — 字典形式为 {"height": int, "width": int}，指定输出图像的尺寸。
• data_format (ChannelDimension 或 str, 可选) — 输出图像的通道维度格式。如果未提供，则从输入图像推断。可以是以下之一：
  • "channels_first" 或 ChannelDimension.FIRST：图像格式为 (num_channels, height, width)。
  • "channels_last" 或 ChannelDimension.LAST：图像格式为 (height, width, num_channels)。
  • "none" 或 ChannelDimension.NONE：图像格式为 (height, width)。
• input_data_format (ChannelDimension 或 str, 可选) — 输入图像的通道维度格式。如果未设置，则从输入图像推断通道维度格式。可以是以下之一：
  • "channels_first" 或 ChannelDimension.FIRST：图像格式为 (num_channels, height, width)。
  • "channels_last" 或 ChannelDimension.LAST：图像格式为 (height, width, num_channels)。
  • "none" 或 ChannelDimension.NONE：图像格式为 (height, width)。

参数

• pixel_values (torch.FloatTensor, 形状为 (batch_size, num_channels, image_size, image_size)) — 对应于输入图像的张量。像素值可以使用 {image_processor_class} 获取。有关详细信息，请参见 {image_processor_class}.__call__（{processor_class} 使用 {image_processor_class} 处理图像）。
• labels (torch.LongTensor, 形状为 (batch_size, sequence_length), 可选) — 用于计算掩码语言建模损失的标签。索引应在 [0, ..., config.vocab_size] 或 -100 之间（参见 input_ids 文档字符串）。索引设置为 -100 的标记将被忽略（掩码），损失只针对标签在 [0, ..., config.vocab_size] 中的标记计算。
• output_attentions (bool, 可选) — 是否返回所有注意力层的注意力张量。有关更多详细信息，请参阅返回张量中的 attentions。
• output_hidden_states (bool, 可选) — 是否返回所有层的隐藏状态。有关更多详细信息，请参阅返回张量中的 hidden_states。
• return_dict (bool, 可选) — 是否返回 ModelOutput 而不是普通元组。

一个 transformers.models.superglue.modeling_superglue.KeypointMatchingOutput 或一个 torch.FloatTensor 元组（如果传入 return_dict=False 或当 config.return_dict=False 时），根据配置 (SuperGlueConfig) 和输入包含各种元素。

• loss (形状为 (1,) 的 torch.FloatTensor，可选) — 训练期间计算的损失。
• matches (torch.FloatTensor, 形状为 (batch_size, 2, num_matches)) — 另一个图像中匹配的关键点索引。
• matching_scores (torch.FloatTensor, 形状为 (batch_size, 2, num_matches)) — 预测匹配的分数。
• keypoints (torch.FloatTensor, 形状为 (batch_size, num_keypoints, 2)) — 给定图像中预测关键点的绝对 (x, y) 坐标。
• mask (torch.IntTensor, 形状为 (batch_size, num_keypoints)) — 指示匹配和匹配分数中的哪些值是关键点匹配信息的掩码。
• hidden_states (tuple[torch.FloatTensor, ...], 可选) — torch.FloatTensor 元组（每个阶段输出一个），形状为 (batch_size, 2, num_channels, num_keypoints)，当传入 output_hidden_states=True 或当 config.output_hidden_states=True 时返回。
• attentions (tuple[torch.FloatTensor, ...], 可选) — torch.FloatTensor 元组（每层一个），形状为 (batch_size, 2, num_heads, num_keypoints, num_keypoints)，当传入 output_attentions=True 或当 config.output_attentions=True 时返回。