Skip to content

ICEORY/VTDNN

Repository files navigation

README

by iceory

since 2020.10.29

Introducion

VTDNN (Visualization Tool for Deep Neural Networks) is a tool for visualizing features and predictions of deep neural networks, we provide web-ui and tkinter-ui for users.

How to use

Web UI

Run the following command

streamlit run app.py

Then open the link on browser

image-20201029205626561

Tkinter-UI

Run the following command

python main.py

Then you will get

image-20201029205844652

Requirements

os: windows10

python packages:
pytorch>=0.4 and torchvision
thinter (if use tkinter-ui)
numpy
matplotlib
PIL
streamlit (if use web-ui)
pydot

About

Visualization tool for deep neural networks

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages