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

cvičení

Konvoluční generátor

Definuj konvoluční generátor podle pokynů DCGAN probíraných v poslední lekci.

torch.nn je pro tvoje pohodlí předimportován jako nn. K dispozici je také vlastní funkce dc_gen_block(), která vrací blok složený z transponované konvoluce, batch normalizace a aktivace ReLU. Tato funkce slouží jako základní stavební prvek konvolučního generátoru. S definicí dc_gen_block() se můžeš seznámit níže.

def dc_gen_block(in_dim, out_dim, kernel_size, stride):
    return nn.Sequential(
        nn.ConvTranspose2d(in_dim, out_dim, kernel_size, stride=stride),
        nn.BatchNorm2d(out_dim),
        nn.ReLU()
    )

Pokyny

100 XP
  • Přidej poslední blok generátoru, který namapuje velikost příznaků na 256.
  • Přidej transponovanou konvoluci s výstupní velikostí 3.
  • Přidej aktivaci tanh.