def __init__(self, output_num_channels=2, input_num_channels=3, inner_nc=128): super(BottleneckFromNet, self).__init__() self.pix2pixSampler = NoSkipPix2PixModel(input_num_channels, output_num_channels, inner_nc=inner_nc)
def __init__(self, output_num_channels=2, input_num_channels=3, inner_nc=512): super(UnwrappedFaceWeightedAverage, self).__init__() self.pix2pixUnwrapped = Pix2PixModel(3) self.pix2pixSampler = NoSkipPix2PixModel(input_num_channels, output_num_channels, inner_nc=inner_nc)