Blok regresoru rámečku
Tvým posledním úkolem je vytvořit regresorový blok pro predikci souřadnic ohraničujícího rámečku. Rozhodneš se použít blok se 2 plně propojenými vrstvami s aktivací ReLU mezi nimi – podobně jako klasifikátor, který jsi definoval/a dříve.
Proměnné vgg_model a input_dim jsou stále k dispozici a torch spolu s torchvision.models jsou naimportované.
Toto cvičení je součástí kurzu
Deep Learning pro obrázky s PyTorchem
Pokyny k cvičení
- Vytvoř proměnnou
num_coordinatess počtem souřadnic ohraničujícího rámečku, které chceš predikovat. - Definuj odpovídající vstupní dimenzi pro první lineární vrstvu a nastav výstupní dimenzi na
32. - Definuj odpovídající výstupní dimenzi v poslední vrstvě regresoru.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# 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, ____),
)