def __init__(self, input_shape, z_size, variational=True, first_kernel=5, first_stride=2, second_kernel=5, second_stride=2):
self.input_shape = input_shape
self.z_size = z_size
encoder = self.Encoder(input_shape, z_size, first_kernel, first_stride, second_kernel, second_stride)
decoder = self.Decoder(z_size, encoder.z_shape, first_kernel, first_stride, second_kernel, second_stride)
BaseVAE.__init__(self, encoder, decoder, variational)
Storeable.__init__(self)
def __init__(self,
input_shape,
z_size,
variational=True,
first_kernel=5,
first_stride=2,
second_kernel=5,
second_stride=2):
self.vae = ConvVAE4Fixed(self, input_shape, z_size, variational,
first_kernel, first_stride, second_kernel,
second_stride)
self.shot_vae = ConvVAE4Fixed(self, input_shape, z_size, variational,
1, first_kernel, first_stride,
second_kernel, second_stride)
Storeable.__init__(self)
def __init__(self,
input_shape,
z_size,
input_channels=3,
output_channels=3,
first_kernel=5,
first_stride=2,
second_kernel=5,
second_stride=2):
self.input_shape = input_shape
self.z_size = z_size
encoder = self.Encoder(input_shape, z_size, input_channels,
first_kernel, first_stride, second_kernel,
second_stride)
decoder = self.Decoder(z_size, encoder.z_shape, output_channels,
first_kernel, first_stride, second_kernel,
second_stride)
BaseAE.__init__(self, encoder, decoder)
Storeable.__init__(self)