Bloc de regresie pentru bounding box
Ultima ta sarcină este să creezi un bloc de regresie pentru a prezice coordonatele bounding box-ului. Vei construi un bloc cu 2 straturi complet conectate și o activare ReLU între ele, similar cu clasificatorul definit anterior.
vgg_model și input_dim sunt disponibile în continuare, iar torch și torchvision.models au fost deja importate.
Acest exercițiu face parte din cursul
Deep Learning pentru imagini cu PyTorch
Instrucțiuni pentru exercițiu
- Creează o variabilă
num_coordinatescu numărul de coordonate ale bounding box-ului de prezis. - Definește dimensiunea de intrare potrivită pentru primul strat liniar și setează dimensiunea de ieșire la
32. - Definește dimensiunea de ieșire potrivită în ultimul strat al regresorului.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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, ____),
)