mobilefacenetpytorch –>onnx –> caffe model | 码农家园
www.codenong.com › cs110919760Dec 09, 2020 · mobilefacenet模型–>onnx–>caffe 一、mobilefacenet–>onnx 训练完成之后,先转为onnx模型并做简化,转换方法可参考我另一篇博客pytorch转onnx、caffe 1. onnx 转换问题 去除模型内部的L2范数归一化。 当我们模型里面有些onnx或其他框架里不支持的算子时,可在转换过程中将其去掉,在cpu里单独实现再衔接上模型的输出( 先确认该OP是否含有参数 ) 1 2 3 4 5 6 7 # 只截取了部分代码, face_model.py MobileFaceNet 类里面,forward 函数下 out = self.conv_6_dw (out) out = self.conv_6_flatten (out)