MXNet
文章平均质量分 96
令狐傻笑
图像算法工程师,主要研究方向为图像分类,目标检测
展开
-
RetinaFace MXNet模型转ONNX转TensorRT
文章目录RetinaFace MXNet模型转ONNX转TensorRT1. github开源代码2. MXNet模型转ONNX模型3. ONNX模型转TensorRT模型3.1 概述3.2 编译3.3 运行4. 推理结果RetinaFace MXNet模型转ONNX转TensorRT1. github开源代码RetinaFace TensorRT推理的开源代码位置在https://github.com/linghu8812/tensorrt_inference/tree/master/RetinaF原创 2020-12-05 09:53:04 · 2272 阅读 · 6 评论 -
MXNet Symbol Batch Normalization fix_gamma=True转ONNX方法
1. 问题当MXNet模型的Batch Normalization的fix_gamma参数为True时,会导致转ONNX模型失败,此时输出的ONNX参数如下图所示,导致ONNX的推理结果和MXNet不一致。2. 解决方法出现MXNet Batch Normalization的fix_gamma参数等于True时,可以手动修改batchnorm_gamma参数值,使ONNX模型输出正常,相关代码如下:sym, arg_params, aux_params = mx.model.load_check原创 2020-11-03 11:13:55 · 484 阅读 · 0 评论