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

Update readme and setup code to be pushed to pip repository.

Browse Source
This commit is contained in:
Alec Helbling
2022-12-31 12:36:15 -05:00
parent 313d1aa650
commit 334662e8c8
11 changed files with 341 additions and 154 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
examples/cnn/cnn.py
View File

@ -3,40 +3,26 @@ from pathlib import Path
from manim import *
from PIL import Image
<<<<<<< HEAD
from manim_ml.neural_network.layers.convolutional3d import Convolutional3DLayer
=======
from manim_ml.neural_network.layers import Convolutional3DLayer
>>>>>>> 0bc3ad561ba224f3d33e9f843665c1d50d64a68b
from manim_ml.neural_network.layers.feed_forward import FeedForwardLayer
from manim_ml.neural_network.layers.image import ImageLayer
from manim_ml.neural_network.neural_network import NeuralNetwork
<<<<<<< HEAD
# Make the specific scene
config.pixel_height = 1200
config.pixel_width = 1900
config.frame_height = 7.0
config.frame_width = 7.0
=======
ROOT_DIR = Path(__file__).parents[2]
>>>>>>> 0bc3ad561ba224f3d33e9f843665c1d50d64a68b
def make_code_snippet():
code_str = """
# Make nn
nn = NeuralNetwork([
<<<<<<< HEAD
ImageLayer(numpy_image, height=1.5),
Convolutional3DLayer(1, 7, 7, 3, 3),
Convolutional3DLayer(3, 5, 5, 3, 3),
Convolutional3DLayer(5, 3, 3, 1, 1),
=======
ImageLayer(numpy_image),
Convolutional3DLayer(3, 3, 3),
Convolutional3DLayer(5, 2, 2),
Convolutional3DLayer(10, 2, 1),
>>>>>>> 0bc3ad561ba224f3d33e9f843665c1d50d64a68b
FeedForwardLayer(3),
FeedForwardLayer(3),
])
@ -64,7 +50,6 @@ class CombinedScene(ThreeDScene):
numpy_image = np.asarray(image)
# Make nn
nn = NeuralNetwork([
<<<<<<< HEAD
ImageLayer(numpy_image, height=1.5),
Convolutional3DLayer(1, 7, 7, 3, 3, filter_spacing=0.32),
Convolutional3DLayer(3, 5, 5, 3, 3, filter_spacing=0.32),
@ -75,16 +60,6 @@ class CombinedScene(ThreeDScene):
layer_spacing=0.25,
)
# Center the nn
=======
ImageLayer(numpy_image, height=3.5),
Convolutional3DLayer(3, 3, 3, filter_spacing=0.2),
Convolutional3DLayer(5, 2, 2, filter_spacing=0.2),
Convolutional3DLayer(10, 2, 1, filter_spacing=0.2),
FeedForwardLayer(3, rectangle_stroke_width=4, node_stroke_width=4).scale(2),
FeedForwardLayer(1, rectangle_stroke_width=4, node_stroke_width=4).scale(2)
], layer_spacing=0.2)
nn.scale(0.9)
>>>>>>> 0bc3ad561ba224f3d33e9f843665c1d50d64a68b
nn.move_to(ORIGIN)
self.add(nn)
# Make code snippet