mirror of
https://github.com/helblazer811/ManimML.git
synced 2025-07-09 19:56:13 +08:00
Used Black to reformat the code in the repository.
This commit is contained in:
@ -9,8 +9,8 @@ config.pixel_width = 1280
|
||||
config.frame_height = 6.0
|
||||
config.frame_width = 6.0
|
||||
|
||||
class ImageNeuralNetworkScene(Scene):
|
||||
|
||||
class ImageNeuralNetworkScene(Scene):
|
||||
def make_code_snippet(self):
|
||||
code_str = """
|
||||
# Make image object
|
||||
@ -32,13 +32,13 @@ class ImageNeuralNetworkScene(Scene):
|
||||
"""
|
||||
|
||||
code = Code(
|
||||
code = code_str,
|
||||
code=code_str,
|
||||
tab_width=4,
|
||||
background_stroke_width=1,
|
||||
background_stroke_color=WHITE,
|
||||
insert_line_no=False,
|
||||
style='monokai',
|
||||
#background="window",
|
||||
style="monokai",
|
||||
# background="window",
|
||||
language="py",
|
||||
)
|
||||
code.scale(0.2)
|
||||
@ -46,22 +46,22 @@ class ImageNeuralNetworkScene(Scene):
|
||||
return code
|
||||
|
||||
def construct(self):
|
||||
image = Image.open('../../tests/images/image.jpeg')
|
||||
image = Image.open("../../tests/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),
|
||||
]
|
||||
nn = NeuralNetwork(layers)
|
||||
nn.scale(0.9)
|
||||
# Center the nn
|
||||
nn.move_to(ORIGIN)
|
||||
nn.rotate(-PI/2)
|
||||
nn.layers[0].image_mobject.rotate(PI/2)
|
||||
nn.rotate(-PI / 2)
|
||||
nn.layers[0].image_mobject.rotate(PI / 2)
|
||||
nn.layers[0].image_mobject.shift([0, -0.4, 0])
|
||||
nn.shift([1.5, 0.3, 0])
|
||||
self.add(nn)
|
||||
@ -71,15 +71,14 @@ class ImageNeuralNetworkScene(Scene):
|
||||
code_snippet.shift([-1.25, 0, 0])
|
||||
self.add(code_snippet)
|
||||
# Play animation
|
||||
self.play(
|
||||
nn.make_forward_pass_animation(run_time=10)
|
||||
)
|
||||
self.play(nn.make_forward_pass_animation(run_time=10))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
"""Render all scenes"""
|
||||
# Feed Forward Neural Network
|
||||
ffnn_scene = FeedForwardNeuralNetworkScene()
|
||||
ffnn_scene.render()
|
||||
# Neural Network
|
||||
# Neural Network
|
||||
nn_scene = NeuralNetworkScene()
|
||||
nn_scene.render()
|
||||
|
Reference in New Issue
Block a user