diff --git a/11-AE/images/ae_reconstructed_epoch_1.png b/11-AE/images/ae_reconstructed_epoch_1.png new file mode 100644 index 0000000..ffe3461 Binary files /dev/null and b/11-AE/images/ae_reconstructed_epoch_1.png differ diff --git a/11-AE/images/ae_reconstructed_epoch_2.png b/11-AE/images/ae_reconstructed_epoch_2.png new file mode 100644 index 0000000..c06dcef Binary files /dev/null and b/11-AE/images/ae_reconstructed_epoch_2.png differ diff --git a/11-AE/images/ae_reconstructed_epoch_3.png b/11-AE/images/ae_reconstructed_epoch_3.png new file mode 100644 index 0000000..1f6c92f Binary files /dev/null and b/11-AE/images/ae_reconstructed_epoch_3.png differ diff --git a/11-AE/images/ae_reconstructed_epoch_4.png b/11-AE/images/ae_reconstructed_epoch_4.png new file mode 100644 index 0000000..fa3b2e8 Binary files /dev/null and b/11-AE/images/ae_reconstructed_epoch_4.png differ diff --git a/11-AE/images/ae_reconstructed_epoch_5.png b/11-AE/images/ae_reconstructed_epoch_5.png new file mode 100644 index 0000000..b4fe099 Binary files /dev/null and b/11-AE/images/ae_reconstructed_epoch_5.png differ diff --git a/11-AE/images/vae_reconstructed_epoch_1.png b/11-AE/images/vae_reconstructed_epoch_1.png deleted file mode 100644 index e52f2d8..0000000 Binary files a/11-AE/images/vae_reconstructed_epoch_1.png and /dev/null differ diff --git a/11-AE/images/vae_reconstructed_epoch_2.png b/11-AE/images/vae_reconstructed_epoch_2.png deleted file mode 100644 index 468d9ef..0000000 Binary files a/11-AE/images/vae_reconstructed_epoch_2.png and /dev/null differ diff --git a/11-AE/images/vae_reconstructed_epoch_3.png b/11-AE/images/vae_reconstructed_epoch_3.png deleted file mode 100644 index e330338..0000000 Binary files a/11-AE/images/vae_reconstructed_epoch_3.png and /dev/null differ diff --git a/11-AE/images/vae_reconstructed_epoch_4.png b/11-AE/images/vae_reconstructed_epoch_4.png deleted file mode 100644 index 5f7c9f0..0000000 Binary files a/11-AE/images/vae_reconstructed_epoch_4.png and /dev/null differ diff --git a/11-AE/images/vae_reconstructed_epoch_5.png b/11-AE/images/vae_reconstructed_epoch_5.png deleted file mode 100644 index 0651019..0000000 Binary files a/11-AE/images/vae_reconstructed_epoch_5.png and /dev/null differ diff --git a/11-AE/main.py b/11-AE/main.py index a1234a7..c90e8d6 100644 --- a/11-AE/main.py +++ b/11-AE/main.py @@ -119,9 +119,9 @@ def call(self, inputs, training=None, mask=None): out = tf.nn.sigmoid(out_logits) # out is just the logits, use sigmoid out = tf.reshape(out, [-1, 28, 28]).numpy() * 255 - x = tf.reshape(x[:batch_size // 2], [-1, 28, 28]) + x = tf.reshape(x[:batch_size // 2], [-1, 28, 28]).numpy() * 255 - x_concat = tf.concat([x, out], axis=0).numpy() * 255. + x_concat = tf.concat([x, out], axis=0).numpy() x_concat = x_concat.astype(np.uint8) index = 0 @@ -132,7 +132,7 @@ def call(self, inputs, training=None, mask=None): new_im.paste(im, (i, j)) index += 1 - new_im.save('images/vae_reconstructed_epoch_%d.png' % (epoch + 1)) + new_im.save('images/ae_reconstructed_epoch_%d.png' % (epoch + 1)) plt.imshow(np.asarray(new_im)) plt.show() print('New images saved !')