ONNXToCaffe:pytorch -> onnx -> caffe,pytorch to caffe,或者其他深度学习框架到onnx和onnx到caffe

时间:2024-06-18 08:45:00
【文件属性】:

文件名称:ONNXToCaffe:pytorch -> onnx -> caffe,pytorch to caffe,或者其他深度学习框架到onnx和onnx到caffe

文件大小:12.32MB

文件格式:ZIP

更新时间:2024-06-18 08:45:00

caffe pytorch onnx Python

代码主要来自 https://github.com/MTlab/onnx2caffe和 https://github.com/205418367/onnx2caffe,感谢他们的贡献。 onnx 到 Caffe 我们可以将 onnx 操作转换为 caffe 层,它不仅来自 ,还来自许多其他 caffe 修改分支,如 ssd-caffe,并且仅支持 onnx opset_version=9。 通过 ONNX 将 pytorch 转换为 Caffe 此工具通过将模型转换为 Caffe 模型仅用于推理 通过 ONNX 将 tensorflow 转换为 Caffe 你可以使用这个 repo 。 其他用于 caffe bt ONNX 的深度学习框架 依赖关系 caffe(支持python) pytorch(如果要转换 onnx,则可选) 昂克斯 运行时 我们建议使用 protobuf


【文件预览】:
ONNXToCaffe-master
----model_generator()
--------__init__.py(0B)
--------broadcast_add.py(1KB)
--------MobileNetV2.py(4KB)
--------broadcast_mul.py(1KB)
--------alexnet.py(4KB)
--------resnet50.py(5KB)
--------googlenet.py(4KB)
--------resnet.py(5KB)
----.gitignore(37B)
----model()
--------README.md(86B)
--------MobileNetV2.onnx(13.84MB)
--------generateONNX.py(690B)
----MyCaffe.py(5KB)
----onnx2caffe()
--------__init__.py(0B)
--------_error_utils.py(2KB)
--------_operators.py(25KB)
--------_weightloader.py(8KB)
--------_graph.py(8KB)
--------_transformers.py(25KB)
----README.md(3KB)
----layerComparator.py(3KB)
----test.py(2KB)
----convertCaffe.py(4KB)
----modelComparator.py(5KB)

网友评论