开始使用免费开始使用

Anchor 生成器

您的团队正在基于 Faster R-CNN 架构并使用预训练骨干网络开发目标检测模型。您的任务是创建锚框,作为参考边界框,用于提出潜在的目标区域。

您将创建 9 个标准锚框(3 种框尺寸和 3 种长宽比)。

已为您导入 torch

本练习是课程的一部分

使用 PyTorch 进行图像深度学习

查看课程

练习说明

  • torchvision.models.detection.rpn 导入 AnchorGenerator
  • 使用 3 个值配置锚框尺寸:((32, 64, 128),)
  • 使用 3 个值配置长宽比:((0.5, 1.0, 2.0),)
  • 使用 anchor_sizesaspect_ratios 实例化 AnchorGenerator

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Import AnchorGenerator
____

# Configure anchor size
anchor_sizes = ____

# Configure aspect ratio
aspect_ratios = ____

# Instantiate AnchorGenerator
rpn_anchor_generator = ____
编辑并运行代码