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

演習

事前学習モデルのバックボーン

R-CNN アーキテクチャを構築しましょう。ここでは、特徴抽出に vgg16 の事前学習済みモデルのバックボーンを使用します。さらに、バックボーンの出力形状を保存しておきます。これは後続のブロック（分類器とボックス回帰器）の入力形状として使われます。

torch、torchvision、torch.nn は nn としてインポート済みです。 モデルは vgg16 として、重みは VGG16_Weights に保存された状態でインポートされています。

指示

100 XP
  • 事前学習済みの VGG16 の重みを読み込みます。
  • .children() を使って classifier の最初のレイヤーを順次ブロックとして参照し、in_features を取り出して input_dim として保存します。
  • features と .children() を使って順次ブロックとしてバックボーンを作成します。
  • バックボーンモデルを表示（print）します。