
个人工具
htshinichi
社畜
展开
-
【caffe】Caffe模型转换为ONNX模型(新版)
在了解了caffe模型的结构和ONNX的结构后,我用python写了一个caffe转onnx的小工具,现只测试了resnet50、alexnet、yolov3的caffe模型和onnx模型推理结果,存在误差,但是在可接受范围内。本工具在转换模型的时候是不需要配置caffe的,只需要安装好protobuf即可。在进行推理测试的时候才需要配置好pycaffe。原创 2019-04-18 09:32:39 · 14658 阅读 · 25 评论 -
【caffe】caffe卷积层和BN层融合
Conv和BN算子融合为什么做融合在神经网络训练的过程中,BN层能够加速网络收敛,并且能够控制过拟合。不过这样也增加了一些运算和参数,在推理过程中,我们可以通过将BN层与卷积层的参数融合,来减少运算,并且为模型稍稍的瘦一下身。依赖protobufnumpy为了省事,使用protobuf来解析caffe模型,因此不需要配置caffe的python接口融合小工具github公式卷...原创 2019-04-11 09:53:02 · 3024 阅读 · 2 评论 -
【ONNX】各深度学习框架的模型转ONNX
文章目录pytorchpytorch安装pytorch转onnx关于pytorch模型的题外话cntkcntk安装cntk转onnxmxnetmxnet安装mxnet转onnxcaffe2caffe2安装caffe2转onnxtensorflowtensorflow安装onnx-tensorflow安装tensorflow转onnxcaffecaffe->caffe2->onnx或ca...原创 2019-04-11 09:17:22 · 18675 阅读 · 5 评论