Блок регресора для рамки
Ваше фінальне завдання — створити блок регресора для передбачення координат обмежувальної рамки. Ви вирішили використати блок із 2 повнозв'язаними шарами з активацією ReLU між ними, подібно до класифікатора, який ви визначали раніше.
Ваші vgg_model та input_dim усе ще доступні, а torch і torchvision.models уже імпортовано.
Ця вправа є частиною курсу
Глибоке навчання для зображень із PyTorch
Інструкції до вправи
- Створіть змінну
num_coordinatesіз кількістю координат обмежувальної рамки для передбачення. - Задайте відповідну розмірність входу для першого лінійного шару та встановіть розмірність виходу
32. - Визначте відповідну розмірність виходу в останньому шарі регресора.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# 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, ____),
)