文件名称:PytorchToCaffe:从Pytorch模型到Caffe模型,支持pytorch 0.3、0.3.1、0.4、0.4.1、1.0、1.0.1、1.2、1.3。请注意,只有pytorch 1.1有一些错误
文件大小:203KB
文件格式:ZIP
更新时间:2024-03-12 05:29:29
Python
由于我们使用的pytorch模型基本上是动态图结构,因此动态图的问题在于,一旦前向不完整,就无法确定图结构,并且caffe是静态图框架,这将导致模型从pytorch转换为caffe到caffe。遇到很多问题,并且pytorch版本迭代非常快,因此不再建议使用此仓库。如果要将pytorch转换为caffe,建议通过此仓库使用pytorch-> onnx-> caffe 。 该代码主要来自 。感谢hahnyuan的贡献。 神经网络工具:转换器和分析器 提供用于pytorch和caffe的神经网络框架的工具。 nn_tools是根据MIT许可证发布的(有关详细信息,请参阅LICENSE文件)。 特征 将pytorch模型转换为caffe模型。 一些方便的工具可以快速操作caffemodel和prototxt(例如获取或设置图层权重)。 支持pytorch版本> = 0.2。(已在0.3,
【文件预览】:
PytorchToCaffe-master
----.gitignore(68B)
----model()
--------__init__.py(2KB)
--------mgn.py(7KB)
--------__pycache__()
--------resnet.py(6KB)
----Pytorch()
--------augmentations.py(19KB)
--------__init__.py(39B)
--------eval.py(2KB)
--------utils.py(1KB)
--------train.py(3KB)
----LICENSE.md(1KB)
----analysis()
--------__init__.py(81B)
--------blob.py(2KB)
--------roi.py(1010B)
--------layers.py(11KB)
--------utils.py(3KB)
--------CaffeA.py(5KB)
--------README.md(95B)
--------PytorchA.py(3KB)
--------__pycache__()
----__init__.py(0B)
----caffe_analyser.py(1KB)
----pytorch_analyser.py(2KB)
----Caffe()
--------__init__.py(0B)
--------ReadMe.md(1KB)
--------caffe.proto(91KB)
--------net.py(97B)
--------caffe_lmdb.py(984B)
--------caffe_net.py(5KB)
--------layer_param.py(6KB)
--------caffe_pb2.py(465KB)
----README.md(6KB)
----example()
--------inceptionv3_pytorch_to_caffe.py(487B)
--------vgg19_pytorch_to_caffe.py(442B)
--------verify_deploy.py(5KB)
--------resnet_pytorch_2_caffe.py(620B)
--------MobileNetV2.py(4KB)
--------mobilenetV2_pytorch_to_caffe.py(644B)
--------alexnet_pytorch_to_caffe.py(443B)
--------__pycache__()
--------Resnet-annalyze.py(2KB)
--------resnet_pytorch_analysis_example.py(340B)
----pytorch_to_caffe.py(29KB)
----001763.jpg(72KB)
----funcs.py(2KB)