Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
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
|
3 years ago | |
|---|---|---|
| models | 3 years ago | |
| ox | 3 years ago | |
| visualization | 4 years ago | |
| .gitignore | 3 years ago | |
| LICENSE | 4 years ago | |
| README.md | 3 years ago | |
| elephant.jpg | 4 years ago | |
| test.py | 3 years 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
