visualtorch

Visualize Model that develope with Pytorch

https://github.com/moyousefipour/visualtorch

Science Score: 44.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
    Found CITATION.cff file
  • codemeta.json file
    Found codemeta.json file
  • .zenodo.json file
    Found .zenodo.json file
  • DOI references
  • Academic publication links
  • Academic email domains
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (11.3%) to scientific vocabulary
Last synced: 6 months ago · JSON representation ·

Repository

Visualize Model that develope with Pytorch

Basic Info
  • Host: GitHub
  • Owner: MoYousefipour
  • License: mit
  • Language: Python
  • Default Branch: master
  • Size: 103 KB
Statistics
  • Stars: 1
  • Watchers: 1
  • Forks: 0
  • Open Issues: 0
  • Releases: 1
Created over 1 year ago · Last pushed 7 months ago
Metadata Files
Readme License Citation

README.md

VisualTorch

Visualtorch is a Python package to help visualize torch (either standalone or included in pytorch) neural network architectures. It allows easy styling to fit most needs. This module supports layered style architecture generation which is great for CNNs (Convolutional Neural Networks), and a graph style architecture, which works great for most models including plain feed-forward networks. For help in citing this project, refer here.

This work is inspired from "paulgavrikov/visualkeras" which is makeing layers from Tensorflow and Keras, and using styles of "HarisIqbal88/PlotNeuralNet" to make it ready to work.

Citation

If you find this project helpful for your research please consider citing it in your publication as follows. @software{Yousefipour_VisualTorch_2025, author = {Yousefipour, Mohammad}, license = {MIT}, month = jul, title = {{VisualTorch}}, url = {https://github.com/MoYousefipour/VisualTorch}, version = {0.1.0}, year = {2025} }

How to Use

Clone the repository.

terminal clone https://github.com/MoYousefipour/VisualTorch.git Change test.py file with you model that written with torch for example for vgg16:

python from torchvision.models import vgg16 import visualtorch import torch.nn as nn model = vgg16() visualtorch.layered_view( model,input_shape=(1,3,224,224),to_file='vgg16.tex',spacing=2,padding=1) Run test.py file, then compile the .tex file that given to see vgg16 architecture.

see VGG16 example!

Owner

  • Name: Mohammad Yousefipour
  • Login: MoYousefipour
  • Kind: user
  • Location: Iran
  • Company: Shahid Chamran University

MS Student of Artificial Intelligence | Shahid Chamran University

Citation (CITATION.cff)

# This CITATION.cff file was generated with cffinit.
# Visit https://bit.ly/cffinit to generate yours today!

cff-version: 1.2.0
title: VisualTorch
message: ' "If you use this software, please cite it as below."'
type: software
authors:
  - given-names: Mohammad
    family-names: Yousefipour
    email: moyousefipour79@gmail.com
    affiliation: >-
      Department of Computer Engineering, Faculty of
      Engineering, Shahid Chamran University of Ahvaz,
      Ahvaz, Iran
    orcid: 'https://orcid.org/0009-0003-6422-6893'
repository-code: 'https://github.com/MoYousefipour/VisualTorch'
license: MIT
commit: e2c5254660b085767825b7c6c894723cfd1f93d3
version: 0.1.0
date-released: '2025-07-25'

GitHub Events

Total
  • Issues event: 1
  • Watch event: 2
  • Push event: 4
  • Public event: 1
Last Year
  • Issues event: 1
  • Watch event: 2
  • Push event: 4
  • Public event: 1