Bloco regressor de caixa
A última coisa que você precisa fazer é criar um bloco regressor pra prever as coordenadas da caixa delimitadora. Você decide ter um bloco com duas camadas totalmente conectadas com uma ativação ReLU no meio, parecido com o classificador que você definiu antes.
Seus e-mails vgg_model e input_dim ainda estão disponíveis, e torch e torchvision.models foram importados.
Este exercício faz parte do curso
Aprendizado profundo para imagens com PyTorch
Instruções do exercício
- Crie uma variável chamada “
num_coordinates” com o número de coordenadas da caixa de limite a ser prevista. - Defina a dimensão de entrada certa para a primeira camada linear e coloque a dimensão de saída como “
32”. - Defina a dimensão de saída certa na última camada do regressor.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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, ____),
)