def train():
from skimage.io import imread
from lasagnekit.misc.plot_weights import grid_plot
x = imread("B_small.png")
x = x / 255.
x = 1 - x
x = x[:, :, 0]
X = x[None, None, :, :]
X = X.astype(np.float32)
w, h, c = X.shape[2], X.shape[3], 1
state = 55
encoder, decoder = build_convnet_deep(w=w, h=h, c=c, size_filters=3, nb_filters=10, nb_layers=3)
encoder = layers_from_list_to_dict(encoder)
decoder = layers_from_list_to_dict(decoder)
input_to_latent_mapping = InputOutputMapping(
[encoder["input"]],
[encoder["z_mean"], encoder["z_log_sigma"]]
)
latent_to_input_mapping = InputOutputMapping(
[decoder["input"]],
[decoder["output"]]
)
learning_rate = 0.001
batch_optimizer = BatchOptimizer(
whole_dataset_in_device=False,
batch_size=1,
max_nb_epochs=10000,
verbose=1,
optimization_procedure=(updates.rmsprop,
{"learning_rate": learning_rate}),
)
# Putting all together
vae = VariationalAutoencoder(input_to_latent_mapping,
latent_to_input_mapping,
batch_optimizer,
rng=rng_mrg.MRG_RandomStreams(seed=state),
input_type=Binary,
X_type=encoder["input"].input_var,
Z_type=decoder["input"].input_var,
nb_z_samples=100)
vae.fit(X)
X = vae.sample(10, only_means=True)
X = X[:, 0]
import matplotlib.pyplot as plt
grid_plot(X, imshow_options={'cmap': 'gray'})
plt.show()
optimization_procedure=(updates.adadelta, {"learning_rate" : 1.}),
batch_size=256,
whole_dataset_in_device=True,
verbose=1)
rng = RandomStreams(seed=1000)
#noise_function = lambda X_batch: X_batch * rng.binomial(size=X_batch.shape, p=0.7)
noise_function = None
model = autoencoder.Autoencoder(nnet_x_to_z, nnet_z_to_x, batch_optimizer, noise_function=noise_function, walkback=1)
model.fit(X)
"""
conv_filters = h.W.get_value()
conv_filters = conv_filters.reshape( (conv_filters.shape[0], conv_filters.shape[2], conv_filters.shape[3]))
grid_plot(conv_filters, imshow_options={"cmap": "gray"})
plt.savefig('out-filters-conv.png')
plt.show()
"""
filters = h.W.get_value().T
filters = get_2d_square_image_view(filters)
grid_plot(filters, imshow_options={"cmap":"gray"})
plt.savefig("out-filters.png")
plt.show()
plt.clf()
samples = model.sample(nb=100, nb_iterations=10000)
samples = get_2d_square_image_view(samples)
grid_plot(samples, imshow_options={"cmap": "gray"})
plt.savefig('out-samples.png')
plt.show()
def test():
state = 10
rng = rng_mrg.MRG_RandomStreams(seed=state)
x_dim = 64
h_dim = 100
x_in = layers.InputLayer((None, x_dim))
l_out = Layer(x_in, num_units=h_dim)
model_encoder = LightweightModel([x_in], [l_out])
h_in = layers.InputLayer((None, h_dim))
l_out = Layer(h_in, num_units=x_dim)
model_decoder = LightweightModel([h_in], [l_out])
h_in = layers.InputLayer((None, 1))
l_out = Layer(h_in, num_units=h_dim)
model_prior = LightweightModel([h_in], [l_out])
def loss_function(model, tensors):
X = tensors["X"]
(h, u), = model_encoder.get_output(X, mode="sample", rng=rng)
#print(X.ndim, h.ndim)
(q_h_given_x, _), = model_encoder.get_output(X, mode="evaluate", on=h)
(p_x_given_h, _), = model_decoder.get_output(h, mode="evaluate", on=X)
ones = T.alloc(np.cast[theano.config.floatX](1.), *h.shape)
zeros = T.alloc(np.cast[theano.config.floatX](0.), *h.shape)
(p_h, _), = model_prior.get_output(zeros[:, 0:1], mode="evaluate", on=ones)
L = -((T.log(p_x_given_h).sum(axis=1) + T.log(p_h).sum(axis=1) - T.log(q_h_given_x).sum(axis=1)))
return (L.mean()), u
input_variables = OrderedDict(
X=dict(tensor_type=T.matrix),
)
functions = dict(
)
# sample function
nb = T.iscalar()
sample_input = T.alloc(np.cast[theano.config.floatX](0.), nb, 1)
from theano.updates import OrderedUpdates
u = OrderedUpdates()
(s_h, u_h), = model_prior.get_output(sample_input, mode="sample", rng=rng)
ones = T.alloc(np.cast[theano.config.floatX](1.), s_h.shape[0], x_dim)
(s_x, u_x), = model_decoder.get_output(s_h, mode="evaluate", on=ones)
sample = theano.function([nb], s_x, updates=u_x)
batch_optimizer = BatchOptimizer(verbose=1, max_nb_epochs=100,
batch_size=256,
optimization_procedure=(updates.rmsprop, {"learning_rate": 0.0001})
)
class Container(object):
def __init__(self, models):
self.models = models
def get_all_params(self):
return [p for m in self.models for p in m.get_all_params()]
models = [model_encoder, model_decoder]
models = Container(models)
darn = Capsule(input_variables,
models,
loss_function,
functions=functions,
batch_optimizer=batch_optimizer)
from sklearn.datasets import load_digits
def build_digits():
digits = load_digits()
X = digits.data
X = X.astype(np.float32) / 16.
return X, digits.images.shape[1:]
X, imshape = build_digits()
darn.fit(X=X)
s=(sample(20))
s = get_2d_square_image_view(s)
grid_plot(s, imshow_options={"cmap": "gray"})
plt.savefig("out.png")
input_variables = OrderedDict(
X=dict(tensor_type=T.matrix),
)
functions = dict(
sample=dict(
get_output=lambda model, X: model.get_output(X, sampler=True)[0],
params=["X"]
),
log_likelihood=dict(
get_output=lambda model, X: T.log(model.get_output(X)[0]).sum(axis=1),
params=["X"]
),
)
batch_optimizer = BatchOptimizer(verbose=1, max_nb_epochs=2,
batch_size=256,
optimization_procedure=(updates.rmsprop, {"learning_rate": 0.001})
)
nade = Capsule(input_variables, model,
loss_function,
functions=functions,
batch_optimizer=batch_optimizer)
nade.fit(X=X)
T = np.ones((100, x_dim)).astype(np.float32)
T = T[:, order]
s = get_2d_square_image_view(nade.sample(T))
grid_plot(s, imshow_options={"cmap": "gray"})
plt.savefig("out.png")
