Backbone model terlatih awal (pre-trained)

Saatnya membangun arsitektur R-CNN! Anda akan menggunakan backbone dari model terlatih awal vgg16 untuk ekstraksi fitur. Anda juga perlu menyimpan bentuk keluaran dari backbone yang akan menjadi bentuk masukan untuk blok-blok berikutnya: classifier dan box regressor.

torch, torchvision, torch.nn sebagai nn telah diimpor. Model telah diimpor sebagai vgg16 dengan bobot disimpan dalam VGG16_Weights.

Latihan ini adalah bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Petunjuk latihan

Muat bobot VGG16 terlatih awal (pre-trained).
Ekstrak in_features dari layer pertama classifier menggunakan .children() sebagai blok sequential dan simpan sebagai input_dim.
Buat backbone sebagai blok sequential menggunakan features dan .children().
Cetak model backbone.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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
____

Edit dan Jalankan Kode