1. 学ぶ
  2. /
  3. コース
  4. /
  5. PyTorch で学ぶ画像向け Deep Learning

Connected

演習

アンカー生成器

あなたのチームは、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 のインスタンスを作成します。