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 618c7a7db8 update 10 months ago
models update 10 months ago
ox update 10 months ago
visualization release 1.0 1 year ago
.gitignore update 10 months ago
LICENSE release 1.0 1 year ago
README.md update 10 months ago
elephant.jpg release 1.0 1 year ago
test.py update 10 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

简介

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.

Python SVG JavaScript CSS Protocol Buffer other