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

演習

バウンディングボックスの予測

物体認識モデルを学習済みです。次は、テスト画像に対してバウンディングボックスとクラス分類を生成したいと考えています。

モデルは model として利用可能で、すでに評価モードになっています。test_image も用意済みで、torch と torch.nn（nn として）がインポートされています。

指示

100 XP
  • test_image をモデルに渡して出力を取得し、output に代入します。
  • 出力から予測されたバウンディングボックスを取り出し、boxes に代入します。
  • 出力から予測スコアを取り出し、scores に代入します。