1. Nauka
    2. /
  2. Kursy
    3. /
  3. Głębokie uczenie dla obrazów z PyTorch
Connected

ćwiczenie

Generator splotowy

Zdefiniuj generator splotowy zgodnie z wytycznymi DCGAN omówionymi w poprzednim filmie.

torch.nn zostało wcześniej zaimportowane jako nn. Dostępna jest również niestandardowa funkcja dc_gen_block(), która zwraca blok złożony z transponowanej konwolucji, normalizacji wsadowej (batch norm) oraz aktywacji ReLU. Funkcja ta stanowi podstawowy element konstrukcji generatora splotowego. Jej definicję znajdziesz poniżej.

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()
    )

Instrukcje

100 XP
  • Dodaj ostatni blok generatora, mapując rozmiar map cech na 256.
  • Dodaj transponowaną konwolucję z rozmiarem wyjściowym 3.
  • Dodaj aktywację tanh.