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

Added GAN visualization.

Browse Source
This commit is contained in:
Alec Helbling
2022-04-22 19:08:28 -04:00
parent ffd31701bf
commit 0152be64b0
24 changed files with 530 additions and 154 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
tests/test_neural_network.py
View File

@ -107,18 +107,18 @@ class NeuralNetworkScene(Scene):
self.play(forward_propagation_animation)
class ImageNeuralNetworkScene(Scene):
class GrayscaleImageNeuralNetworkScene(Scene):
def construct(self):
image = Image.open('images/image.jpeg')
numpy_image = np.asarray(image)
# Make nn
layers = [
ImageLayer(numpy_image, height=1.4),
FeedForwardLayer(3),
FeedForwardLayer(3),
FeedForwardLayer(5),
FeedForwardLayer(3),
FeedForwardLayer(6)
FeedForwardLayer(6),
ImageLayer(numpy_image, height=1.4)
]
nn = NeuralNetwork(layers)
nn.scale(1.3)
@ -129,6 +129,27 @@ class ImageNeuralNetworkScene(Scene):
self.play(nn.make_forward_pass_animation(run_time=5))
self.play(nn.make_forward_pass_animation(run_time=5))
class ImageNeuralNetworkScene(Scene):
def construct(self):
image = Image.open('../assets/gan/real_image.jpg')
numpy_image = np.asarray(image)
# Make nn
layers = [
FeedForwardLayer(3),
FeedForwardLayer(5),
FeedForwardLayer(3),
FeedForwardLayer(6),
ImageLayer(numpy_image, height=1.4)
]
nn = NeuralNetwork(layers)
nn.scale(1.3)
# Center the nn
nn.move_to(ORIGIN)
self.add(nn)
# Play animation
self.play(nn.make_forward_pass_animation(run_time=5))
self.play(nn.make_forward_pass_animation(run_time=5))
class EmbeddingNNScene(Scene):
@ -174,7 +195,7 @@ class LayerRemovalScene(Scene):
image = Image.open('images/image.jpeg')
numpy_image = np.asarray(image)
layer = FeedForwardLayer(5),
layer = FeedForwardLayer(5)
layers = [
ImageLayer(numpy_image, height=1.4),
FeedForwardLayer(3),
@ -186,7 +207,34 @@ class LayerRemovalScene(Scene):
nn = NeuralNetwork(layers)
self.play(Create(nn))
self.play(nn.remove_layer(layer))
remove_animation = nn.remove_layer(layer)
print("before remove")
self.play(remove_animation)
print(nn)
print("after remove")
class LayerInsertionScene(Scene):
def construct(self):
image = Image.open('images/image.jpeg')
numpy_image = np.asarray(image)
layers = [
ImageLayer(numpy_image, height=1.4),
FeedForwardLayer(3),
FeedForwardLayer(3),
FeedForwardLayer(6)
]
nn = NeuralNetwork(layers)
self.play(Create(nn))
layer = FeedForwardLayer(5)
insert_animation = nn.insert_layer(layer, 4)
self.play(insert_animation)
print(nn)
print("after remove")
if __name__ == "__main__":
"""Render all scenes"""