Sınıflandırıcı bloğu

Sıradaki görevin, orijinal VGG16 sınıflandırıcısının yerini alacak bir sınıflandırıcı bloğu oluşturmak. Araya bir ReLU aktivasyonu koyarak iki tam bağlantılı katmandan oluşan bir blok kullanmaya karar verdin.

Son egzersizde tanımladığın vgg_model ve input_dim çalışma alanında mevcut ve torch ile torchvision.models 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ı

Yalnızca kedi ve köpek tespiti yaptığını varsayarak sınıf sayısını içeren num_classes adlı bir değişken oluştur.
nn.Sequential kullanarak sıralı bir blok oluştur.
in_features değeri input_dim olacak şekilde bir doğrusal katman oluştur.
Çıkış özelliklerini sınıflandırıcının son katmanına ekle.

Uygulamalı interaktif egzersiz

Bu örnek kodu tamamlayarak bu egzersizi bitirin.

# Create a variable with the number of classes
____
    
# Create a sequential block
classifier = ____(
	# Create a linear layer with input features
	____(____, 512),
	nn.ReLU(),
	# Add the output dimension to the classifier
	nn.Linear(512, ____),
)

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