opensource/ManimML
1
0
Fork 0
mirror of https://github.com/helblazer811/ManimML.git synced 2025-07-15 07:57:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Used Black to reformat the code in the repository.

Browse Source
This commit is contained in:
Alec Helbling
2023-01-01 23:24:59 -05:00
parent 334662e8c8
commit 3d6e8072e1
71 changed files with 1701 additions and 1135 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
examples/interpolation/interpolation.py
View File

@ -1,7 +1,7 @@
"""Visualization of VAE Interpolation"""
import sys
import os
sys.path.append(os.environ["PROJECT_ROOT"])
from manim import *
import pickle
@ -12,22 +12,25 @@ import examples.variational_autoencoder.variational_autoencoder as variational_a
"""
The VAE Scene for the twitter video.
"""
config.pixel_height = 720
config.pixel_width = 1280
config.pixel_height = 720
config.pixel_width = 1280
config.frame_height = 6.0
config.frame_width = 6.0
# Set random seed so point distribution is constant
np.random.seed(1)
class InterpolationScene(MovingCameraScene):
"""Scene object for a Variational Autoencoder and Autoencoder"""
def construct(self):
# Set Scene config
vae = variational_autoencoder.VariationalAutoencoder(dot_radius=0.035, layer_spacing=0.5)
vae = variational_autoencoder.VariationalAutoencoder(
dot_radius=0.035, layer_spacing=0.5
)
vae.move_to(ORIGIN)
vae.encoder.shift(LEFT*0.5)
vae.decoder.shift(RIGHT*0.5)
vae.encoder.shift(LEFT * 0.5)
vae.decoder.shift(RIGHT * 0.5)
mnist_image_handler = variational_autoencoder.MNISTImageHandler()
image_pair = mnist_image_handler.image_pairs[3]
# Make forward pass animation and DO NOT run it
@ -35,23 +38,22 @@ class InterpolationScene(MovingCameraScene):
# Make the interpolation animation
interpolation_images = mnist_image_handler.interpolation_images
interpolation_animation = vae.make_interpolation_animation(interpolation_images)
embedding_zoom_animation = self.camera.auto_zoom([
vae.embedding,
vae.decoder,
vae.output_image
], margin=0.5)
embedding_zoom_animation = self.camera.auto_zoom(
[vae.embedding, vae.decoder, vae.output_image], margin=0.5
)
# Make animations
forward_pass_animations = []
for i in range(7):
anim = vae.decoder.make_forward_propagation_animation(run_time=0.5)
forward_pass_animations.append(anim)
forward_pass_animation_group = AnimationGroup(*forward_pass_animations, lag_ratio=1.0)
forward_pass_animation_group = AnimationGroup(
*forward_pass_animations, lag_ratio=1.0
)
# Make forward pass animations
self.play(Create(vae), run_time=1.5)
self.play(FadeOut(vae.encoder), run_time=1.0)
self.play(embedding_zoom_animation, run_time=1.5)
interpolation_animation = AnimationGroup(
forward_pass_animation_group,
interpolation_animation
forward_pass_animation_group, interpolation_animation
)
self.play(interpolation_animation, run_time=9.0)