
神经网络
文章平均质量分 55
令狐傻笑
图像算法工程师,主要研究方向为图像分类,目标检测
展开
-
Ubuntu18.04 配置mmdetection实现Faster RCNN目标检测
1.安装mmcv1.1 安装方法方法1pip install mmcv-full -f ttps://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html方法2pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html方法3git clone https://g原创 2021-05-22 10:26:47 · 406 阅读 · 1 评论 -
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 · 554 阅读 · 0 评论 -
PyTorch学习——Andrew Ng machine-learning-ex4 Neural Networks Learning
目录Exercise 4: Neural Networks Learning1. Neural Networks模型2. 加载数据3. 网络训练4.结果比较4.1 激活函数比较4.2 隐藏层个数比较Exercise 4: Neural Networks Learning需要用到的库import numpy as npimport scipy.ioi...原创 2019-05-03 23:14:20 · 513 阅读 · 0 评论