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_sizes和aspect_ratios实例化AnchorGenerator。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Import AnchorGenerator
____
# Configure anchor size
anchor_sizes = ____
# Configure aspect ratio
aspect_ratios = ____
# Instantiate AnchorGenerator
rpn_anchor_generator = ____