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

Reformatted the code using black, allowd for different orientation NNs, made an option for highlighting the active filter in a CNN forward pass.

Browse Source
This commit is contained in:
Alec Helbling
2023-01-09 15:52:37 +09:00
parent 39b0b133ce
commit ba63116b37
19 changed files with 485 additions and 283 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
manim_ml/gridded_rectangle.py
View File

@ -24,6 +24,7 @@ class GriddedRectangle(VGroup):
):
super().__init__()
# Fields
self.color = color
self.mark_paths_closed = mark_paths_closed
self.close_new_points = close_new_points
self.grid_xstep = grid_xstep
@ -33,8 +34,6 @@ class GriddedRectangle(VGroup):
self.grid_stroke_opacity = grid_stroke_opacity if show_grid_lines else 0.0
self.stroke_width = stroke_width
self.rotation_angles = [0, 0, 0]
self.rectangle_width = width
self.rectangle_height = height
self.show_grid_lines = show_grid_lines
# Make rectangle
self.rectangle = Rectangle(
@ -129,3 +128,9 @@ class GriddedRectangle(VGroup):
normal_vector = np.cross((vertex_1 - vertex_2), (vertex_1 - vertex_3))
return normal_vector
def set_color(self, color):
"""Sets the color of the gridded rectangle"""
self.color = color
self.rectangle.set_color(color)
self.rectangle.set_stroke_color(color)