繪製邊界框
已經準備好影像與邊界框的張量,現在你可以把方框畫在影像上,並以視覺方式檢查其準確度。
已匯入 torch、torchvision、torchvision.transforms。影像已轉為名為 image_tensor 的張量。座標已指定給變數:x_min、y_min、x_max、y_max。
本練習屬於課程
使用 PyTorch 進行影像深度學習
練習說明
- 從
torchvision.utils匯入draw_bounding_boxes。 - 定義邊界框
bbox,為包含x_min、y_min、x_max、y_max的清單。 - 將
image_tensor與bbox_tensor傳入draw_bounding_boxes,把方框畫在影像上,並將輸出指定給img_bbox。 - 把
img_bbox張量轉成影像。
動手互動練習
試著完成這個範例程式碼,體驗一下這個練習。
# Import draw_bounding_boxes
____
# Define the bounding box coordinates
bbox = ____
bbox_tensor = torch.tensor(bbox).unsqueeze(0)
# Implement draw_bounding_boxes
img_bbox = ____(____, ____, width=3, colors="red")
# Tranform tensors to image
transform = transforms.Compose([
____
])
plt.imshow(transform(img_bbox))
plt.show()