Bloque de regresor de cajas
Tu tarea final es crear un bloque regresor para predecir las coordenadas de los cuadros delimitadores. Decides usar un bloque con 2 capas totalmente conectadas con una activación ReLU entre ellas, similar al clasificador que definiste antes.
Tus vgg_model e input_dim siguen disponibles y ya se han importado torch y torchvision.models.
Este ejercicio forma parte del curso
Deep Learning para imágenes con PyTorch
Instrucciones del ejercicio
- Crea una variable
num_coordinatescon el número de coordenadas del cuadro a predecir. - Define la dimensión de entrada adecuada para la primera capa lineal y establece la dimensión de salida en
32. - Define la dimensión de salida adecuada en la última capa del regresor.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# 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, ____),
)