Blok regresor kotak

Tugas terakhir Anda adalah membuat blok regresor untuk memprediksi koordinat bounding box. Anda memutuskan untuk menggunakan blok dengan 2 lapisan fully connected dengan aktivasi ReLU di antaranya, mirip dengan classifier yang Anda definisikan sebelumnya.

vgg_model dan input_dim Anda masih tersedia dan torch serta torchvision.models telah diimpor.

Latihan ini merupakan bagian dari kursus

Deep Learning untuk Gambar dengan PyTorch

Lihat Kursus

Instruksi latihan

Buat variabel num_coordinates dengan jumlah koordinat bounding box yang akan diprediksi.
Definisikan dimensi masukan yang sesuai untuk lapisan linear pertama dan atur dimensi keluarannya menjadi 32.
Definisikan dimensi keluaran yang sesuai pada lapisan terakhir regresor.

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

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

Edit dan Jalankan Kode