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.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
Guangyao Chen 618c7a7db8 update 8 months ago
models update 8 months ago
ox update 8 months ago
visualization release 1.0 1 year ago
.gitignore update 8 months ago
LICENSE release 1.0 1 year ago
README.md update 8 months ago
elephant.jpg release 1.0 1 year ago
test.py update 8 months ago

README.md

open-exchange

open-exchange 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

You can also choose to visit the http://dgx.pkuml.org:6269/ website directly. After uploading the JSON file of IR, you can see the basic information about the model.

Model

Framework ResNet50 Inception V3 ShuffleNet FCN LSTM
TensorFlow
PyTorch

Test for Tensorflow and PyTorch

CUDA_VSIBLE_DEVICES=0 python test.py