You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
Guangyao Chen 8965fb58a6 update 5 months ago
docs update 5 months ago
models update 6 months ago
ox update 6 months ago
visualization update 6 months ago
.gitignore update 6 months ago
LICENSE update 6 months ago
README.md update 6 months ago
elephant.jpg update 6 months ago
requirements.txt update 5 months ago
setup.py update 5 months ago
test.py update 6 months ago

README.md

Model-Pivot

Model-Pivot is a model conversion and visualization tool to help users inter-operate among different deep learning frameworks. Convert models between PyTorch and Tensorflow.
IR is based on the National Information Technology Standardization ```Neural Network Representation and Model Compression Part 1: Convolution Neural Network``.

Requirments

  • tensorflow==1.8.0
  • pytorch==0.4.0
  • torchvision==0.2.0
  • protobuf>=3.6.1
  • python>=3.6
  • flask

How to deploy visualization on Web

If you want to access the deployed web page from an external network, you should first modify the host and port for the ./visualization/app.py file.

You can deploy it on Web by running:

python app.py

Model

Framework ResNet50 Inception V3 ShuffleNet FCN LSTM
TensorFlow
PyTorch

Test for Tensorflow and PyTorch

CUDA_VSIBLE_DEVICES=0 python test.py

简介

Model-Pivot is a model conversion and visualization tool to help users inter-operate among different deep learning frameworks. Convert models between PyTorch and Tensorflow.

https://model-pivot.readthedocs.io

Python SVG JavaScript CSS Protocol Buffer other