邊界框預測
你已經訓練好一個物件辨識模型。現在，你想用它為測試影像產生邊界框與分類結果。
模型已可透過 model 取得，且已設定為評估模式。test_image 也已提供，並且已匯入 torch 與 torch.nn（作為 nn）。
本練習屬於課程
使用 PyTorch 進行影像深度學習
練習說明
- 取得
test_image的模型輸出，並指定給output。 - 從輸出中擷取預測的邊界框，並指定給
boxes。 - 從輸出中擷取預測的分數，並指定給
scores。
動手互動練習
試著完成這個範例程式碼，體驗一下這個練習。
# Get model's prediction
with torch.no_grad():
output = ____
# Extract boxes from the output
boxes = ____
# Extract scores from the output
scores = ____
print(boxes, scores)