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

Final animations

Browse Source
This commit is contained in:
Alec Helbling
2022-02-16 22:01:53 -05:00
committed by Alec Helbling
parent 92d8a3d59a
commit e8857acc83
4 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
examples/VAEScene.mp4
View File

Binary file not shown.

BIN
final_videos/TestNeuralNetworkScene.mp4
View File

Binary file not shown.

BIN
final_videos/VAEScene.mp4
View File

Binary file not shown.

32
src/vae.py
View File

@ -6,21 +6,20 @@ and Traditional Autoencoders.
""" """
from configparser import Interpolation from configparser import Interpolation
from random import sample from random import sample
from typing_extensions import runtime
from manim import * from manim import *
import pickle import pickle
import numpy as np import numpy as np
import neural_network import neural_network
from scipy.interpolate import make_interp_spline from scipy.interpolate import make_interp_spline
class VariationalAutoencoder(Group): class VariationalAutoencoder(VGroup):
"""Variational Autoencoder Manim Visualization""" """Variational Autoencoder Manim Visualization"""
def __init__( def __init__(
self, encoder_nodes_per_layer=[5, 3], decoder_nodes_per_layer=[3, 5], point_color=BLUE, self, encoder_nodes_per_layer=[5, 3], decoder_nodes_per_layer=[3, 5], point_color=BLUE,
dot_radius=0.05, ellipse_stroke_width=2.0 dot_radius=0.05, ellipse_stroke_width=2.0
): ):
super(Group, self).__init__() super(VGroup, self).__init__()
self.encoder_nodes_per_layer = encoder_nodes_per_layer self.encoder_nodes_per_layer = encoder_nodes_per_layer
self.decoder_nodes_per_layer = decoder_nodes_per_layer self.decoder_nodes_per_layer = decoder_nodes_per_layer
self.point_color = point_color self.point_color = point_color
@ -78,7 +77,7 @@ class VariationalAutoencoder(Group):
embedding.add(self.point_dots) embedding.add(self.point_dots)
return embedding return embedding
def _construct_image_mobject(self, input_image, height=2): def _construct_image_mobject(self, input_image, height=2.3):
"""Constructs an ImageMobject from a numpy grayscale image""" """Constructs an ImageMobject from a numpy grayscale image"""
# Convert image to rgb # Convert image to rgb
input_image = np.repeat(input_image, 3, axis=0) input_image = np.repeat(input_image, 3, axis=0)
@ -153,7 +152,7 @@ class VariationalAutoencoder(Group):
self.input_image.move_to(self.encoder.get_left()) self.input_image.move_to(self.encoder.get_left())
self.input_image.shift(LEFT) self.input_image.shift(LEFT)
self.output_image.move_to(self.decoder.get_right()) self.output_image.move_to(self.decoder.get_right())
self.output_image.shift(RIGHT * 1.2) self.output_image.shift(RIGHT)
# Make encoder forward pass # Make encoder forward pass
encoder_forward_pass = self.encoder.make_forward_propagation_animation(run_time=per_unit_runtime) encoder_forward_pass = self.encoder.make_forward_propagation_animation(run_time=per_unit_runtime)
# Make red dot in embedding # Make red dot in embedding
@ -202,25 +201,26 @@ class VariationalAutoencoder(Group):
"""Makes an animation interpolation""" """Makes an animation interpolation"""
num_images = len(interpolation_images) num_images = len(interpolation_images)
# Make madeup path # Make madeup path
interpolation_latent_path = np.linspace([-0.5, -1], [1, 1.3], num=num_images) interpolation_latent_path = np.linspace([-0.7, -1.2], [1.2, 1.5], num=num_images)
# Make the path animation # Make the path animation
first_dot_location = self.embedding.axes.coords_to_point(*interpolation_latent_path[0]) first_dot_location = self.embedding.axes.coords_to_point(*interpolation_latent_path[0])
last_dot_location = self.embedding.axes.coords_to_point(*interpolation_latent_path[-1])
moving_dot = Dot(first_dot_location, radius=self.dot_radius, color=RED) moving_dot = Dot(first_dot_location, radius=self.dot_radius, color=RED)
animation_list = [GrowFromCenter(moving_dot)] self.add(moving_dot)
animation_list = [Create(Line(first_dot_location, last_dot_location, color=RED), run_time=0.1*num_images)]
for image_index in range(num_images - 1): for image_index in range(num_images - 1):
next_index = image_index + 1 next_index = image_index + 1
# Get path # Get path
next_point = interpolation_latent_path[next_index] next_point = interpolation_latent_path[next_index]
next_position = self.embedding.axes.coords_to_point(*next_point) next_position = self.embedding.axes.coords_to_point(*next_point)
# Draw path from current point to next point # Draw path from current point to next point
move_animation = moving_dot.animate.move_to(next_position) move_animation = moving_dot.animate(run_time=0.1*num_images).move_to(next_position)
animation_list.append(move_animation) animation_list.append(move_animation)
interpolation_animation = Succession(*animation_list, run_time=0.1*num_images) interpolation_animation = AnimationGroup(*animation_list)
# Make the images animation # Make the images animation
animation_list = [] animation_list = [Wait(0.5)]
for numpy_image in interpolation_images: for numpy_image in interpolation_images:
numpy_image = interpolation_images[0]
numpy_image = numpy_image[None, :, :] numpy_image = numpy_image[None, :, :]
manim_image = self._construct_image_mobject(numpy_image) manim_image = self._construct_image_mobject(numpy_image)
# Move the image to the correct location # Move the image to the correct location
@ -228,10 +228,10 @@ class VariationalAutoencoder(Group):
# Add the image # Add the image
animation_list.append(FadeIn(manim_image, run_time=0.1)) animation_list.append(FadeIn(manim_image, run_time=0.1))
# Wait # Wait
animation_list.append(Wait(1 / frame_rate)) # animation_list.append(Wait(1 / frame_rate))
# Remove the image # Remove the image
animation_list.append(FadeOut(manim_image, run_time=0.1)) # animation_list.append(FadeOut(manim_image, run_time=0.1))
images_animation = Succession(*animation_list) images_animation = AnimationGroup(*animation_list, lag_ratio=1.0)
# Combine the two into an AnimationGroup # Combine the two into an AnimationGroup
animation_group = AnimationGroup( animation_group = AnimationGroup(
interpolation_animation, interpolation_animation,
@ -285,8 +285,8 @@ class VAEScene(Scene):
mnist_image_handler = MNISTImageHandler() mnist_image_handler = MNISTImageHandler()
image_pair = mnist_image_handler.image_pairs[3] image_pair = mnist_image_handler.image_pairs[3]
vae.move_to(ORIGIN) vae.move_to(ORIGIN)
vae.scale(1.2) vae.scale(1.3)
self.add(vae) self.play(Create(vae), run_time=3)
# Make a forward pass animation # Make a forward pass animation
forward_pass_animation = vae.make_forward_pass_animation(image_pair) forward_pass_animation = vae.make_forward_pass_animation(image_pair)
self.play(forward_pass_animation) self.play(forward_pass_animation)