GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → MTLab → onnx2caffe

MTLab / onnx2caffe

Licence: MIT license
pytorch to caffe by onnx

Programming Languages

python
139335 projects - #7 most used programming language

Labels

caffepytorchonnx

Projects that are alternatives of or similar to onnx2caffe

X2paddle
Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)
Stars: ✭ 315 (-7.62%)
Mutual labels:  caffe, onnx
Mmdnn
MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.
Stars: ✭ 5,472 (+1504.69%)
Mutual labels:  caffe, onnx
Php Opencv Examples
Tutorial for computer vision and machine learning in PHP 7/8 by opencv (installation + examples + documentation)
Stars: ✭ 333 (-2.35%)
Mutual labels:  caffe, onnx
yolov5 onnx2caffe
yolov5 onnx caffe
Stars: ✭ 73 (-78.59%)
Mutual labels:  caffe, onnx
Keras Oneclassanomalydetection
[5 FPS - 150 FPS] Learning Deep Features for One-Class Classification (AnomalyDetection). Corresponds RaspberryPi3. Convert to Tensorflow, ONNX, Caffe, PyTorch. Implementation by Python + OpenVINO/Tensorflow Lite.
Stars: ✭ 102 (-70.09%)
Mutual labels:  caffe, onnx
Onnx2caffe
pytorch to caffe by onnx
Stars: ✭ 282 (-17.3%)
Mutual labels:  caffe, onnx
Visualdl
Deep Learning Visualization Toolkit（『飞桨』深度学习可视化工具 ）
Stars: ✭ 4,258 (+1148.68%)
Mutual labels:  caffe, onnx
ppq
PPL Quantization Tool (PPQ) is a powerful offline neural network quantization tool.
Stars: ✭ 281 (-17.6%)
Mutual labels:  caffe, onnx
Onnx Chainer
Add-on package for ONNX format support in Chainer
Stars: ✭ 83 (-75.66%)
Mutual labels:  caffe, onnx
Deepo
Setup and customize deep learning environment in seconds.
Stars: ✭ 6,145 (+1702.05%)
Mutual labels:  caffe, onnx
arcface retinaface mxnet2onnx
arcface and retinaface model convert mxnet to onnx.
Stars: ✭ 53 (-84.46%)
Mutual labels:  caffe, onnx
Php Opencv
php wrapper for opencv
Stars: ✭ 194 (-43.11%)
Mutual labels:  caffe, onnx
Centerx
This repo is implemented based on detectron2 and centernet
Stars: ✭ 403 (+18.18%)
Mutual labels:  caffe, onnx
Pinto model zoo
A repository that shares tuning results of trained models generated by TensorFlow / Keras. Post-training quantization (Weight Quantization, Integer Quantization, Full Integer Quantization, Float16 Quantization), Quantization-aware training. TensorFlow Lite. OpenVINO. CoreML. TensorFlow.js. TF-TRT. MediaPipe. ONNX. [.tflite,.h5,.pb,saved_model,tfjs,tftrt,mlmodel,.xml/.bin, .onnx]
Stars: ✭ 634 (+85.92%)
Mutual labels:  caffe, onnx
Ncnn
ncnn is a high-performance neural network inference framework optimized for the mobile platform
Stars: ✭ 13,376 (+3822.58%)
Mutual labels:  caffe, onnx
Netron
Visualizer for neural network, deep learning, and machine learning models
Stars: ✭ 17,193 (+4941.94%)
Mutual labels:  caffe, onnx
torch-model-compression
针对pytorch模型的自动化模型结构分析和修改工具集，包含自动分析模型结构的模型压缩算法库
Stars: ✭ 126 (-63.05%)
Mutual labels:  onnx
tensorrt-ssd-easy
No description or website provided.
Stars: ✭ 32 (-90.62%)
Mutual labels:  caffe
GuidedNet
Caffe implementation for "Guided Optical Flow Learning"
Stars: ✭ 28 (-91.79%)
Mutual labels:  caffe
SWCaffe
A Deep Learning Framework customized for Sunway TaihuLight
Stars: ✭ 37 (-89.15%)
Mutual labels:  caffe
View All Similar Projects ➔

Convert pytorch to Caffe by ONNX

This tool converts pytorch model to Caffe model by ONNX
only use for inference

Dependencies

  • caffe (with python support)
  • pytorch 0.4 (optional if you only want to convert onnx)
  • onnx

we recomand using protobuf 2.6.1 and install onnx from source

git clone --recursive https://github.com/onnx/onnx.git
cd onnx 
python setup.py install

How to use

run test.py to make sure it has been installed correctly
To convert onnx model to caffe:

python convertCaffe.py ./model/MobileNetV2.onnx ./model/MobileNetV2.prototxt ./model/MobileNetV2.caffemodel

Current support operation

  • Conv
  • ConvTranspose
  • BatchNormalization
  • MaxPool
  • AveragePool
  • Relu
  • Sigmoid
  • Dropout
  • Gemm (InnerProduct only)
  • Add
  • Mul
  • Reshape
  • Upsample
  • Concat
  • Flatten

TODO List

  • support all onnx operations (which is impossible)
  • merge batchnormization to convolution
  • merge scale to convolution
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].