Skip to content
Snippets Groups Projects
Commit db4eb673 authored by Mattia Mancini's avatar Mattia Mancini
Browse files

Add download and compress data

parent 5885ab5a
No related branches found
No related tags found
No related merge requests found
Expand all Show whitespace changes
Inline Side-by-side
Untitled.ipynb deleted 100644 → 0
+ 0
− 198
View file @ 5885ab5a
%% Cell type:code id:9cd4d4bf tags:
``` python
!pip install matplotlib
import numpy as np
from tqdm import tqdm_notebook
import matplotlib.pyplot as plt
%matplotlib inline
import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
tf.config.experimental.set_memory_growth(gpu, True)
import h5py
dataset=None
timerange=None
freqrange=None
ids=None
fpath = '/project/lofarsw/Data/Dynspec/dset.h5'
with h5py.File(fpath, 'r') as fin:
in_dataset=fin['/data'][:]
in_timerange=fin['/time_range'][:]
in_freqrange=fin['/freq_range'][:]
in_ids=fin['timestamps'][:]
g = np.where(~np.isnan(in_dataset.sum(axis=(1,2))))
dataset = in_dataset[g]
timerange= in_timerange[g]
freqrange= in_freqrange[g]
ids = in_ids[g]
#avg = dataset[:, :, 404]
import time
from IPython import display
import os
class ConvVarAutoencoder(tf.keras.Model):
def __init__(self, latent_dim, imsize):
super().__init__(self)
self.latent_dim = latent_dim
shape = imsize
self.encoder = tf.keras.Sequential(
[
tf.keras.layers.InputLayer(input_shape=imsize),
tf.keras.layers.Conv2D(filters=32, kernel_size=(3,3), strides=(2, 2), activation='relu'),
#tf.keras.layers.MaxPool2D(pool_size=(3,3), strides=(2, 2)),
tf.keras.layers.Conv2D(filters=32, kernel_size=(3,3), strides=(2, 2), activation='relu'),
tf.keras.layers.Conv2D(filters=32, kernel_size=(3,3), strides=(2, 2), activation='relu'),
tf.keras.layers.Flatten(),
# No activation
tf.keras.layers.Dense(latent_dim + latent_dim),
]
)
rescaled = (shape[0] // 4, shape[1] // 4)
self.decoder = tf.keras.Sequential(
[
tf.keras.layers.InputLayer(input_shape=(latent_dim,)),
tf.keras.layers.Dense(units=rescaled[0]*rescaled[1]*32, activation=tf.nn.relu),
tf.keras.layers.Reshape(target_shape=(rescaled[0], rescaled[1], 32)),
tf.keras.layers.Conv2DTranspose(
filters=32, kernel_size=(3,3), strides=2, padding='same',
activation='relu'),
tf.keras.layers.Conv2DTranspose(
filters=32, kernel_size=(3,3), strides=2, padding='same',
activation='relu'),
# No activation
#tf.keras.layers.UpSampling2D(size=(3, 3), interpolation='nearest'),
tf.keras.layers.Conv2DTranspose(
filters=1, kernel_size=(3,3), strides=1, padding='same'),
]
)
@tf.function
def sample(self, eps=None):
if eps is None:
eps = tf.random.normal(shape=(100, self.latent_dim))
return self.decode(eps, apply_sigmoid=True)
def encode(self, x):
mean, logvar = tf.split(self.encoder(x), num_or_size_splits=2, axis=1)
return mean, logvar
def reparameterize(self, mean, logvar):
eps = tf.random.normal(shape=mean.shape)
return eps * tf.exp(logvar * .5) + mean
def decode(self, z, apply_sigmoid=False):
logits = self.decoder(z)
if apply_sigmoid:
probs = tf.sigmoid(logits)
return probs
return logits
optimizer = tf.keras.optimizers.Adam(1e-4)
def log_normal_pdf(sample, mean, logvar, raxis=1):
log2pi = tf.math.log(2. * np.pi)
return tf.reduce_sum(
-.5 * ((sample - mean) ** 2. * tf.exp(-logvar) + logvar + log2pi),
axis=raxis)
def compute_loss(model, x):
mean, logvar = model.encode(x[0])
z = model.reparameterize(mean, logvar)
x_logit = model.decode(z)
cross_ent = tf.nn.sigmoid_cross_entropy_with_logits(logits=x_logit, labels=x[1])
logpx_z = -tf.reduce_sum(cross_ent, axis=[1, 2, 3])
logpz = log_normal_pdf(z, 0., 0.)
logqz_x = log_normal_pdf(z, mean, logvar)
return -tf.reduce_mean(logpx_z + logpz - logqz_x)
@tf.function
def train_step(model, x, optimizer):
"""Executes one training step and returns the loss.
This function computes the loss and gradients, and uses the latter to
update the model's parameters.
"""
with tf.GradientTape() as tape:
loss = compute_loss(model, x)
gradients = tape.gradient(loss, model.trainable_variables)
optimizer.apply_gradients(zip(gradients, model.trainable_variables))
epochs = 200
num_examples_to_generate = 16
# keeping the random vector constant for generation (prediction) so
# it will be easier to see the improvement.
random_vector_for_generation = tf.random.normal(
shape=[num_examples_to_generate, latent_dim])
model = ConvVarAutoencoder(latent_dim, training_dset.shape[1:])
os.makedirs('test', exist_ok=True)
def generate_and_save_images(model, epoch, test_sample):
mean, logvar = model.encode(test_sample)
z = model.reparameterize(mean, logvar)
predictions = model.sample(z)
fig = plt.figure(figsize=(16, 16))
for i in range(0, 8):
plt.subplot(4, 4, i + 1)
plt.imshow(np.array(predictions[i, :, :, 0]).T, cmap='viridis', origin='lower', aspect='auto', vmin=0, vmax=1)
plt.xlabel('time')
plt.ylabel('freq')
for i in range(0, 8):
plt.subplot(4, 4, i + 9)
plt.imshow(np.array(test_sample[i, :, :, 0]).T, cmap='viridis', origin='lower', aspect='auto', vmin=0, vmax=1)
plt.xlabel('freq')
plt.ylabel('time')
# tight_layout minimizes the overlap between 2 sub-plots
plt.savefig('test/image_at_epoch_{:04d}.png'.format(epoch))
plt.show()
assert batch_size >= num_examples_to_generate
for test_batch in test_dataset.take(1):
test_sample = test_batch[0][0:num_examples_to_generate, :, :, :]
```
download_and_compress_data.ipynb 0 → 100644
+ 8241
− 0
View file @ db4eb673
This diff is collapsed.
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment