Regressionsblock för bounding box
Din sista uppgift är att skapa ett regressionsblock för att förutsäga bounding box-koordinater. Du väljer ett block med 2 fullt anslutna lager med en ReLU-aktivering emellan, liknande den klassificerare du definierade tidigare.
Din vgg_model och input_dim är fortfarande tillgängliga, och torch samt torchvision.models har importerats.
Den här övningen är en del av kursen
Djupinlärning för bilder med PyTorch
Övningsinstruktioner
- Skapa en variabel
num_coordinatesmed antalet bounding box-koordinater som ska förutsägas. - Definiera lämplig indatadimension för det första linjära lagret och sätt utdatadimensionen till
32. - Definiera lämplig utdatadimension i regressionsblockets sista lager.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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, ____),
)