Bloque regresor de caja
Tu tarea final consiste en crear un bloque regresor para predecir las coordenadas del cuadro delimitador. Decides crear un bloque con dos capas totalmente conectadas con una activación ReLU entre ellas, similar al clasificador que definiste anteriormente.
Tus direcciones de correo electrónico vgg_model y input_dim siguen disponibles, y las direcciones torch y torchvision.models se han importado.
Este ejercicio forma parte del curso
Aprendizaje profundo para imágenes con PyTorch
Instrucciones del ejercicio
- Crea una variable
num_coordinatescon el número de coordenadas del cuadro delimitador que se va 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 completando 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, ____),
)