ashishpatel26/Tools-to-Design-or-Visualize-Architecture-of-Neural-Network

Public

Tools to Design or Visualize Architecture of Neural Network

  • Stars5.4k+4 tuần này
  • Forks647
  • Watchers71
  • Issues5
  • Ngày tạo06/11/2019
  • Cập nhật gần nhất01/08/2025
  • Ngôn ngữ—
  • Giấy phép—
README.md34.4 KBRaw

Tools to Design or Visualize Architecture of Neural Network

Net2Vis automatically generates abstract visualizations for convolutional neural networks from Keras code.

Visualkeras is a Python package to help visualize Keras (either standalone or included in tensorflow) neural network architectures. It allows easy styling to fit most needs. As of now it supports layered style architecture generation which is great for CNNs (Convolutional Neural Networks) and a grap style architecture.

import visualkeras

model = ...

visualkeras.layered_view(model).show() # display using your system viewer
visualkeras.layered_view(model, to_file='output.png') # write to disk
visualkeras.layered_view(model, to_file='output.png').show() # write and show

visualkeras.layered_view(model)

Visualization of the VGG16 neural network with stacked layers of decreasing size.

Python script for illustrating Convolutional Neural Network (ConvNet)

img

AlexNet style

enter image description here

enter image description here

Latex code for drawing neural networks for reports and presentation. Have a look into examples to see how they are made. Additionally, lets consolidate any improvements that you make and fix any bugs to help more people with this code.

img

img

TensorBoard’s Graphs dashboard is a powerful tool for examining your TensorFlow model.

enter image description here

In Caffe you can use caffe/draw.py to draw the NetParameter protobuffer:

enter image description here

enter image description here

enter image description here

A library for Keras for investigating architectures and parameters of sequential models.

VGG 16 Architecture

           OPERATION           DATA DIMENSIONS   WEIGHTS(N)   WEIGHTS(%)

              Input   #####      3  224  224
         InputLayer     |   -------------------         0     0.0%
                      #####      3  224  224
      Convolution2D    \|/  -------------------      1792     0.0%
               relu   #####     64  224  224
      Convolution2D    \|/  -------------------     36928     0.0%
               relu   #####     64  224  224
       MaxPooling2D   Y max -------------------         0     0.0%
                      #####     64  112  112
      Convolution2D    \|/  -------------------     73856     0.1%
               relu   #####    128  112  112
      Convolution2D    \|/  -------------------    147584     0.1%
               relu   #####    128  112  112
       MaxPooling2D   Y max -------------------         0     0.0%
                      #####    128   56   56
      Convolution2D    \|/  -------------------    295168     0.2%
               relu   #####    256   56   56
      Convolution2D    \|/  -------------------    590080     0.4%
               relu   #####    256   56   56
      Convolution2D    \|/  -------------------    590080     0.4%
               relu   #####    256   56   56
       MaxPooling2D   Y max -------------------         0     0.0%
                      #####    256   28   28
      Convolution2D    \|/  -------------------   1180160     0.9%
               relu   #####    512   28   28
      Convolution2D    \|/  -------------------   2359808     1.

Nội dung README thuộc về ashishpatel26 và cộng đồng đóng góp, giấy phép Không có giấy phép. Nguồn: github.com/ashishpatel26/Tools-to-Design-or-Visualize-Architecture-of-Neural-Network