cyclegan - Serbipunk/notes GitHub Wiki

G inference

model (CycleGANModel) .net_GA (models.networks.UnetGenerator)

model.set_input # load image

model.test()

BaseModel.test() -> forward()

    def forward(self):
        self.fake_B = self.netG_A(self.real_A)  # G_A(A)
        self.rec_A = self.netG_B(self.fake_B)   # G_B(G_A(A))
        self.fake_A = self.netG_B(self.real_B)  # G_B(B)
        self.rec_B = self.netG_A(self.fake_A)   # G_A(G_B(B))

compute_visuals()

null function

model.get_current_visuals()

return Tensors

save_images()