Kutu regresörü bloğu
Son görevin, sınır kutusu koordinatlarını tahmin edecek bir regresör bloğu oluşturmak. Daha önce tanımladığın sınıflandırıcıya benzer şekilde, arada bir ReLU etkinleştirmesi olan 2 tam bağlı katmanlı bir blok kullanmaya karar verdin.
vgg_model ve input_dim hâlâ kullanılabilir durumda ve torch ile torchvision.models içe aktarılmıştır.
Bu egzersiz
PyTorch ile Görüntüler için Deep Learning
kursunun bir parçasıdırEgzersiz talimatları
- Tahmin edilecek sınır kutusu koordinasyonu sayısını içeren
num_coordinatesadlı bir değişken oluştur. - İlk doğrusal katman için uygun giriş boyutunu tanımla ve çıkış boyutunu
32olarak ayarla. - Regresörün son katmanında uygun çıkış boyutunu tanımla.
Uygulamalı interaktif egzersiz
Bu örnek kodu tamamlayarak bu egzersizi bitirin.
# Define the number of coordinates
____
bb = nn.Sequential(
# Add input and output dimensions
nn.Linear(____, ____),
nn.ReLU(),
# Add the output for the last regression layer
nn.Linear(32, ____),
)