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

cvičení

Konvoluční diskriminátor

Teď, když máš generátor DCGAN připravený, zbývá před trénováním ještě definovat konvoluční diskriminátor.

torch.nn je importován pod obvyklým aliasem. K sestavení konvolučního diskriminátoru použiješ vlastní funkci gc_disc_block(), která vrací blok složený z konvoluce, batch normalizace a aktivační funkce Leaky ReLU. Definici funkce dc_disc_block() najdeš níže.

def dc_disc_block(in_dim, out_dim, kernel_size, stride):
    return nn.Sequential(
        nn.Conv2d(in_dim, out_dim, kernel_size, stride=stride),
        nn.BatchNorm2d(out_dim),
        nn.LeakyReLU(0.2),
    )

Pokyny

100 XP
  • Přidej první blok diskriminátoru pomocí vlastní funkce dc_disc_block() se 3 vstupními příznakových mapami a 512 výstupními příznakovými mapami.
  • Přidej konvoluční vrstvu s výstupní velikostí 1.
  • V metodě forward() předej vstup přes sekvenční blok definovaný v __init__().