insightface之mxnet model转caffemodel

本文详细介绍了如何使用MXNet2Caffe工具将Insightface的LResNet100E-IR模型从MXNet格式转换为Caffe格式,包括将JSON模型转换为PROTOXT文件,并修改PROTOXT以匹配Caffe层,最终生成CAFFEMODEL文件的过程。
摘要由CSDN通过智能技术生成

insightface之mxnet model转caffemodel

使用的工具是更新后的MXNet2Caffe:https://github.com/GarrickLin/MXNet2Caffe

mxnet model转caffemodel的思路:先将mxnet的model-symbol.json转化为caffe的prototxt,再由model-symbol.json、model-0000.params以及生成的prototxt文件转化为caffemodel。

这里演示insightface的LResNet100E-IR的网络及模型的转化,模型效果经过测试基本没多少损失:

1、将LResNet100E-IR-symbol.json以及LResNet100E-IR-0154.params放置到model_mxnet文件夹下面。
2、终端运行下述命令:
python json2prototxt.py --mx-json model_mxnet/LResNet100E-IR-symbol.json --cf-prototxt model_caffe/LResNet100E-IR.prototxt
能够直接将LResNet100E-IR-symbol.json转化为LResNet100E-IR.prototxt,但会提示忽略掉未知的层,其实这些未知的层对于生成的LResNet100E-IR.prototxt网络性能几乎没有影响。

评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值