BaşlayınÜcretsiz Başlayın

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

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

kursunun bir parçasıdır
Kursu Görüntüle

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ı interaktif egzersiz

Bu örnek kodu tamamlayarak bu egzersizi bitirin.

# 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