開始使用免費開始

錨點產生器

你的團隊正在以 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 = ____
編輯並執行程式碼