ćwiczenie

Blok regresora ramki

Twoim ostatnim zadaniem jest stworzenie bloku regresora do przewidywania współrzędnych ramki ograniczającej. Zdecydujesz się na blok z 2 warstwami w pełni połączonymi i aktywacją ReLU między nimi – podobnie jak w zdefiniowanym wcześniej klasyfikatorze.

Model vgg_model oraz zmienna input_dim są nadal dostępne, a biblioteki torch i torchvision.models zostały już zaimportowane.

Instrukcje

100 XP

Utwórz zmienną num_coordinates przechowującą liczbę współrzędnych ramki ograniczającej do przewidzenia.
Zdefiniuj odpowiedni wymiar wejściowy dla pierwszej warstwy liniowej i ustaw wymiar wyjściowy na 32.
Zdefiniuj odpowiedni wymiar wyjściowy w ostatniej warstwie regresora.

.css-6su6fj{-webkit-flex-shrink:0;-ms-flex-negative:0;flex-shrink:0;}ćwiczenie

Instrukcje

ćwiczenie