Sınırlayıcı kutu tahmini

Bir nesne tanıma modeli eğittin. Şimdi, bir test görüntüsü için sınırlayıcı kutular ve sınıflandırmalar üretmek istiyorsun.

Model model olarak hazır ve değerlendirme modunda. test_image da mevcut ve torch ile torch.nn (kısaca nn) içe aktarılmış durumda.

Bu egzersiz, kursun bir parçasıdır

PyTorch ile Görüntüler için Deep Learning

Kursa Göz Atın

Egzersiz talimatları

test_image için modelin çıktısını al ve output değişkenine ata.
Çıktıdan tahmin edilen sınırlayıcı kutuları çıkar ve boxes değişkenine ata.
Çıktıdan tahmin edilen skorları çıkar ve scores değişkenine ata.

Uygulamalı etkileşimli egzersiz

Bu egzersizi bu örnek kodu tamamlayarak deneyin.

# Get model's prediction
with torch.no_grad():
    output = ____

# Extract boxes from the output
boxes = ____

# Extract scores from the output
scores = ____

print(boxes, scores)

Kodu Düzenle ve Çalıştır