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

cvičení

Generátor

Generátor sítě GAN přijímá na vstupu náhodný šumový vektor a produkuje vygenerovaný obrázek. Aby byla jeho architektura co nejznovupoužitelnější, předáš jak vstupní, tak výstupní tvar jako parametry modelu. Díky tomu budeš moci stejný model použít s různými velikostmi vstupního šumu i obrázků různých rozměrů.

torch.nn je již naimportovaný jako nn. Máš k dispozici také vlastní funkci gen_block(), která vrací blok skládající se z: lineární vrstvy, batch normalizace a aktivační funkce ReLU. Použiješ ji jako stavební prvek generátoru.

def gen_block(in_dim, out_dim):
    return nn.Sequential(
        nn.Linear(in_dim, out_dim),
        nn.BatchNorm1d(out_dim),
        nn.ReLU(inplace=True)
    )

Pokyny

100 XP
  • Definuj self.generator jako sekvenční model.
  • Za poslední gen_block přidej lineární vrstvu s odpovídající velikostí vstupu a výstupní velikostí out_dim.
  • Za lineární vrstvu přidej sigmoidovou aktivaci.
  • V metodě forward() předej vstup modelu přes self.generator.