Önceden eğitilmiş model omurgası

Bir R-CNN mimarisi kurma zamanı! Özellik çıkarımı için vgg16 önceden eğitilmiş modelinin omurgasını kullanacaksın. Ayrıca, omurganın çıktı şeklini saklamayı unutma; bu değer sonraki bloklar olan sınıflandırıcı ve kutu regresörü için girdi şekli olacak.

torch, torchvision, torch.nn (nn olarak) içe aktarılmış durumda. Model vgg16 olarak, ağırlıklar ise VGG16_Weights içinde içe aktarılmıştır.

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ı

Önceden eğitilmiş VGG16 ağırlıklarını yükle.
classifier'ın ilk katmanından .children() ile ardışık bir blok olarak in_features değerini çıkar ve input_dim olarak sakla.
features ve .children() kullanarak ardışık bir blok şeklinde bir omurga (backbone) oluştur.
Omurga modelini yazdır.

Uygulamalı interaktif egzersiz

Bu örnek kodu tamamlayarak bu egzersizi bitirin.

# Load pretrained weights
vgg_model = vgg16(weights=____)

# Extract the input dimension
input_dim = nn.Sequential(*list(vgg_model.classifier.____()))[0].____

# Create a backbone with convolutional layers
backbone = nn.Sequential(*list(____))

# Print the backbone model
____

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