关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>

Guangyao Chen 8965fb58a6 update		2 年前
docs	update	2 年前

models	update	2 年前

ox	update	2 年前

visualization	update	2 年前

.gitignore	update	2 年前

LICENSE	update	2 年前

README.md	update	2 年前

elephant.jpg	update	2 年前

requirements.txt	update	2 年前

setup.py	update	2 年前

test.py	update	2 年前

README.md

Model-Pivot

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

pytorch tensorflow model-exchange 模型转换工具

Python SVG JavaScript CSS Protocol Buffer other