MXNet从上手到入门
分享
文章平均质量分 58
MXNet 是亚马逊（Amazon）选择的深度学习库，并且也许是最优秀的库 。
本栏目会通过几个方向的实验来讲述MXNet的使用，一个入门级的教程，希望可以推动MXNet的普及。
超级会员免费看
穆友航
有事可电邮：muyouhang@gmail.com
展开
-
【MXNet C++】人脸检测与口罩识别（提供c++源码、mxnet模型文件、win演示程序安装包）
一个用以检测戴口罩人脸的工程，基于MXNet、OpenCV开源框架，使用C++语言实现。工程代码：待更新结果展示：主函数#include "FR.h"#include "FaceAlignment.h"#include "Funcation.h"int main() { FR fr; Funcation fc; FaceAlignment fa; bool VIS = ...原创 2020-02-12 14:55:35 · 4623 阅读 · 17 评论 -
【MXNet Gluon】模型训练使用多块显卡加速（multi-gpu)
使用单块显卡时的代码： devices = mx.gpu(0) data = mx.nd.array(batch_data).as_in_context(devices) label = mx.nd.array(batch_label).as_in_context(devices) # 更新生成器G with autograd.record(): ou...原创 2019-01-07 15:58:20 · 3236 阅读 · 2 评论 -
【MXNet C++】使用预训练模型进行特征提取
已编译的release库： MXNet_win64_cuda8.0_c++_release_lib使用样例：FeatureExtractor FE;vector<double> feature = FE.Extract("test.jpg");FeatureExtractor类 代码如下：#pragma once#include <iost原创 2018-12-18 08:46:25 · 1020 阅读 · 1 评论 -
【MXNet Gluon】使用预训练好的模型fine-tune
finetune关键代码prenet=ResNet(466)net=ResNet(3400)ctx = [mx.gpu(i) for i in range(3)]if finetune ==1: prenet.load_params('params/net-%d.params' % (start_iter),ctx) #features为需要保留的模型参数，outpu...原创 2018-05-07 20:22:19 · 2814 阅读 · 2 评论 -
【MXNet Gluon】自己动手实现fit函数，实现断点保存
【MXNet Gluon】自己动手实现断点保存……用过caffe的炼丹师应该都知道，在用caffe训练模型时，可以通过命令行窗口提前终止训练过程，caffe会自动保存当前状态的参数，以供继续训练。 ……但是，对于MXNet，无论你使用何种接口，都不存在这种机制。 ……在这里，提供一个实现方式。 实现该过程，分两个步骤： ……一是写自己的训练过程函数 ……二是监控模型训练过程中来自命令...原创 2018-07-29 22:15:51 · 965 阅读 · 0 评论 -
【MXNet Symbol】手动实现数据迭代器之读取图像对
代码此代码用以读取图像对，可以用于图像分割、GAN等要求数据与标签都是图像的任务中。也可以用于多模态图像的应用。需要注意的是：不同类型数据对的文件名要一直，数据也要一致，否则会出现数据不匹配的错误，导致错误的实验结果。# -*- coding:utf-8 -*-import osimport cv2import numpy as npimport randomdef getImgL...原创 2018-12-10 00:40:00 · 299 阅读 · 0 评论 -
【MXNet Gluon】基于斯坦福狗的品种分类数据集训练SSD检测模型
本文所使用的数据和模型可在CSDN资源页下载。 链接： 网络定义文件 数据链接和检测用lst文件 本文主要对原来的代码进行了整理，方便调用和训练。 主要参考了Gluon的SSD例子。1. SSD网络模型定义ssd.pyimport mxnet as mximport matplotlib.pyplot as pltimport os.path as ospimp...原创 2017-12-28 19:26:13 · 3526 阅读 · 2 评论 -
【MXNet TX2】成功在TX2上编译MXNet源码，并使用GPU计算（踩坑与填坑）
【MXNet TX2】成功在TX2上编译MXNet源码，并使用GPU计算（踩坑与填坑）首先，便是去看官网的教程了，按照官网教程走，肯定可以成功编译mxnet，并可以执行python代码：import mxnet as mxmx.gpu()但是，当你尝试执行gpu计算任务时，会提示 no kernel image **之类的错误。例如：Check failed: (err) =...原创 2018-08-03 23:04:43 · 2651 阅读 · 0 评论 -
【MXNet Gluon】目标检测数据集制作+SSD 模型训练
基于MXNet gluon 的SSD模型训练官方教程链接：SSD 使用Gluon1. 数据准备车辆检测数据集（友情提示1积分，如无积分。可私信） 请下载上述数据集，内部包含图片+工具代码+lst文件例子。2. 数据集制作官方文档中是这样写的： 但是，在实际操作过程中我发现，A需要设置为4才好用。具体，可以参考 数据集里的文件。3.模型训练3.1 训练代码 通过修改data_shap原创 2017-08-30 23:47:37 · 9857 阅读 · 24 评论 -
【MXNet Symbol】 多指标评价模型性能
MXNet 分类模型训练之采用多指标评价模型能力（accuracy,cross-entropy,top_k_accuracy）原创 2017-09-06 16:55:29 · 1596 阅读 · 0 评论 -
【MXNet Symbol】 访问模型权值参数
关于这个的用途，主要是分析模型内部的参数，可视化一系列权重信息。Step 1. 加载模型model= resnet34_v2(classes=1000)model.load_parameters('model.params',ctx=mx.cpu(0))或者net_name='resnet'image_size=224sym, arg_params, aux_params = mx...原创 2019-05-19 14:10:07 · 484 阅读 · 1 评论