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

Got image mobject to conv3d transformation working (not great, but passable)

Browse Source
This commit is contained in:
Alec Helbling
2022-12-29 22:45:15 -05:00
parent 245ce9424b
commit ff4c1ffded
2 changed files with 49 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
manim_ml/gridded_rectangle.py
View File

@ -20,6 +20,8 @@ class GriddedRectangle(VGroup):
self.grid_stroke_opacity = grid_stroke_opacity self.grid_stroke_opacity = grid_stroke_opacity
self.stroke_width = stroke_width self.stroke_width = stroke_width
self.rotation_angles = [0, 0, 0] self.rotation_angles = [0, 0, 0]
self.rectangle_width = width
self.rectangle_height = height
# Make rectangle # Make rectangle
self.rectangle = Rectangle( self.rectangle = Rectangle(
width=width, width=width,

49
manim_ml/neural_network/layers/image_to_convolutional3d.py
View File

@ -46,6 +46,52 @@ class ImageToConvolutional3DLayer(VGroupNeuralNetworkLayer, ThreeDLayer):
animations = [] animations = []
image_mobject = self.input_layer.image_mobject image_mobject = self.input_layer.image_mobject
target_feature_map = self.output_layer.feature_maps[0] target_feature_map = self.output_layer.feature_maps[0]
# Map image mobject to feature map
# Make rotation of image
x_rotation = ApplyMethod(
image_mobject.rotate,
ThreeDLayer.three_d_x_rotation,
[1, 0, 0],
image_mobject.get_center(),
run_time=0.5
)
y_rotation = ApplyMethod(
image_mobject.rotate,
ThreeDLayer.three_d_y_rotation,
[0, 1, 0],
image_mobject.get_center(),
run_time=0.5
)
# Set opacity
set_opacity = ApplyMethod(
image_mobject.set_opacity,
0.2,
run_time=0.5
)
# Scale the max of width or height to the
# width of the feature_map
max_width_height = max(image_mobject.width, image_mobject.height)
scale_factor = target_feature_map.rectangle_width / max_width_height
scale_image = ApplyMethod(
image_mobject.scale,
scale_factor,
run_time=0.5
)
# Move the image
move_image = ApplyMethod(
image_mobject.move_to,
target_feature_map
)
# Compose the animations
animation = Succession(
x_rotation,
y_rotation,
scale_image,
set_opacity,
move_image,
)
return animation
"""
# Make the object 3D by adding it back into camera frame # Make the object 3D by adding it back into camera frame
def remove_fixed_func(image_mobject): def remove_fixed_func(image_mobject):
# self.camera.remove_fixed_orientation_mobjects(image_mobject) # self.camera.remove_fixed_orientation_mobjects(image_mobject)
@ -71,10 +117,9 @@ class ImageToConvolutional3DLayer(VGroupNeuralNetworkLayer, ThreeDLayer):
image_mobject image_mobject
) )
animations.append(make_fixed) animations.append(make_fixed)
return AnimationGroup()
return AnimationGroup(*animations) return AnimationGroup(*animations)
"""
def scale(self, scale_factor, **kwargs): def scale(self, scale_factor, **kwargs):
super().scale(scale_factor, **kwargs) super().scale(scale_factor, **kwargs)