錨點產生器
你的團隊正在以 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 = ____