Giter VIP home page Giter VIP logo

arcface_retinaface_mxnet2onnx's Introduction

Update2 models

arcface 提取码:jks7
retinaface 提取码:otx7

Update1 fix_gamma

retinaface_mnet025_v1
retinaface_mnet025_v2
In mxnet symbol, BN has fix_gamma, if fix_gamma is true, then set gamma to 1 and its gradient to 0, you can find this in mxnet API.
In retinaface_mnet025_v1, fix_gamma in 'conv_3_dw_batchnorm' is true,but its value is 0.000007107922556315316(you can see weight by Netron).However, forward mxnet model, the gamma of 'conv_3_dw_batchnorm' is 1.This bug may cause mxnet output is different from onnx model.

fix bn gamma model have upload(/Retinaface/retinaface_mnet025_v1, /Retinaface/retinaface_mnet025_v2).

Update

Retinaface fixed softmax bug.
Upsample is implemented using Resize.
Upsample is implemented using ConvTranspose.

arcface_retinaface_mxnet2onnx

arcface and retinaface model convert mxnet to onnx

environment

Ubuntu 18.04
MxNet 1.5.0
onnx 1.7.0 (protobuf 3.0.0)
onnxruntime 1.3.0
Python 3.6.9
cv2 3.3.1

tested models

arcface:from ZQCNN mobilefacenet-res2-6-10-2-dim512
retinaface:mnet.25
arcface/model-r34-amf-slim and retinaface-R50 also can convert successfully, but model file is too big to upload.

arcface

Insightface中ArcFace MxNet2ONNX踩坑

retinaface

Insightface中Retinaface MxNet2ONNX踩坑

Results

arcface onnx_mxnet_output
retinaface onnx_mxnet_output

reference

insightface

arcface_retinaface_mxnet2onnx's People

Contributors

zheshipinyinmc avatar

Stargazers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.