1. Learn
    2. /
  2. 课程
    3. /
  3. 使用 PyTorch 进行图像深度学习
Connected

道练习

Anchor 生成器

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

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

已为您导入 torch。

说明

100 XP
  • 从 torchvision.models.detection.rpn 导入 AnchorGenerator。
  • 使用 3 个值配置锚框尺寸：((32, 64, 128),)。
  • 使用 3 个值配置长宽比：((0.5, 1.0, 2.0),)。
  • 使用 anchor_sizes 和 aspect_ratios 实例化 AnchorGenerator。