LeNet Style View

LeNet Style View module for pytorch model visualization.

visualtorch.lenet_style.lenet_view(model, input_shape, input_dtype=None, to_file=None, min_z=1, min_xy=10, max_xy=2000, scale_z=1, scale_xy=1, type_ignore=None, color_map=None, palette='okabe_ito', low_dim_orientation='z', background_fill='white', padding=10, spacing=10, draw_funnel=True, shade_step=10, font=None, font_color='black', opacity=255, max_channels=100, offset_z=10, level_gap=None, show_dimension=True, show_input=True, outline_width=1, connector_fill=None, connector_width=1, one_dim_orientation=None)

Generate a LeNet style architecture visualization for a given torch model.

Return type:

<module ‘PIL.Image’ from ‘/home/docs/checkouts/readthedocs.org/user_builds/visualtorch/envs/latest/lib/python3.11/site-packages/PIL/Image.py’>

Deprecated:

Since version 1.4.0. Use visualtorch.render(model, input_shape, style=”lenet”, …) instead - every parameter here is forwarded unchanged.