def __call__(self, x): """Discriminate images. """ with nn.parameter_scope("discriminator"): h = self.from_RGB(x, self.resolution_list[-1], self.channel_list[-1]) for i in range(1, len(self.resolution_list))[::-1]: h = self.cnn(h, self.resolution_list[i], self.channel_list[i], self.channel_list[i - 1]) h = minibatch_stddev(h) h = self.last_cnn(h, self.resolution_list[0], self.resolution_list[0], self.resolution_list[0]) return h
def transition(self, x, alpha): """Generator in the transition period, almost the same as the callable. """ with nn.parameter_scope("discriminator"): h = self.transition_cnn(x, self.resolution_list[-2], self.resolution_list[-1], self.channel_list[-2], self.channel_list[-1], alpha) for i in range(1, len(self.resolution_list) - 1)[::-1]: h = self.cnn(h, self.resolution_list[i], self.channel_list[i], self.channel_list[i - 1]) h = minibatch_stddev(h) h = self.last_cnn(h, self.resolution_list[0], self.resolution_list[0], self.resolution_list[0]) return h