1. Learn
    2. /
  2. Courses
    3. /
  3. PyTorch で学ぶ画像向け Deep Learning
Connected

Exercise

バウンディングボックスを描画する

画像テンソルとバウンディングボックスのテンソルを用意できたので、画像の上にボックスを描画して、精度を目視で確認します。

torch、torchvision、torchvision.transforms はインポート済みです。画像はすでに image_tensor としてテンソルに変換されています。座標は x_min、y_min、x_max、y_max の各変数に代入されています。

Instructions

100 XP
  • torchvision.utils から draw_bounding_boxes をインポートします。
  • x_min、y_min、x_max、y_max からなるリストとしてバウンディングボックス bbox を定義します。
  • image_tensor と bbox_tensor を draw_bounding_boxes に渡して画像上にボックスを描画し、出力を img_bbox に代入します。
  • img_bbox テンソルを画像に変換します。