Phxmark / arcface_retinaface_mxnet2caffe

arcface and retinaface model convert mxnet to caffe, merge bn and scale into conv/fc.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

arcface_retinaface_mxnet2caffe

arcface and retinaface model convert mxnet to caffe.
Merge BN and Scale into Conv/FC.

Operator Support Lists

Convolution
ChannelwiseConvolution
BatchNorm
Activation
ElementWiseSum
_Plus
Concat
Crop
Pooling
Flatten
FullyConnected
SoftmaxOutput&SoftmaxFocalOutput
SoftmaxActivation
LeakyReLU
elemwise_add
UpSampling
Deconvolution
Clip
Reshape
SoftmaxActivation
Dropout
softmax

Tested models

arcface/mobilefacenet-res2-6-10-2-dim512 from ZQCNN
arcface/model-r34-amf-slim
retinaface/mnet.25
retinaface/R50

Merge BN and Scale into Conv/FC

Caffe模型合并BN和Scale到Conv/FC层

Bugs

R50 Maxpool, caffe use ceil to compute the output shape. Mxnet can use ceil and floor.

Reference

advanced-mxnet2caffe

About

arcface and retinaface model convert mxnet to caffe, merge bn and scale into conv/fc.


Languages

Language:Python 100.0%