1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Deep Learning pro obrázky s PyTorchem
Connected

cvičení

Blok regresoru rámečku

Tvým posledním úkolem je vytvořit regresorový blok pro predikci souřadnic ohraničujícího rámečku. Rozhodneš se použít blok se 2 plně propojenými vrstvami s aktivací ReLU mezi nimi – podobně jako klasifikátor, který jsi definoval/a dříve.

Proměnné vgg_model a input_dim jsou stále k dispozici a torch spolu s torchvision.models jsou naimportované.

Pokyny

100 XP
  • Vytvoř proměnnou num_coordinates s počtem souřadnic ohraničujícího rámečku, které chceš predikovat.
  • Definuj odpovídající vstupní dimenzi pro první lineární vrstvu a nastav výstupní dimenzi na 32.
  • Definuj odpovídající výstupní dimenzi v poslední vrstvě regresoru.