mxnet
_rookie_coder
Talk is cheap.Show me the code!
展开
-
mxnet学习(5):模型参数
模型参数选择class mxnet.gluon.Block(prefix = None, params = None)其中collect_params()是该类下面的一个成员函数,而该类是所有的网络和模型的基类。该函数返回一个参数字典包含了这个Block和其所有孩子的参数,同样可以有选择性的返回部分参数,使用正则表达式来选择。eg:model.collect_params('conv1_we...原创 2019-03-19 15:04:56 · 2070 阅读 · 0 评论 -
maxnet学习(1):image函数
reference:https://mxnet.incubator.apache.org/api/python/image/image.html#mxnet.image.imread注意这里不是gluon的函数mxnet.image.imdecode和mxnet.image.imread的区别二者都是使用C++的opencv对图像进行处理,imdecode将图片解码为NDarray,而在...原创 2019-03-15 16:57:58 · 4694 阅读 · 0 评论 -
mxnet学习(2):Tensor和Normalize
reference:https://mxnet.incubator.apache.org/api/python/gluon/data.htmlvision下面主要有一些数据集和transformstransforms可以在训练数据的时候用来做数据增强,这里的数据集增强和image API中的数据集增强大同小异。需要注意哪些转换需要在ToTensor之前,哪些需要在ToTensor之后Tak...原创 2019-03-15 17:19:46 · 2765 阅读 · 0 评论 -
mxnet学习(3):autograd
reference:https://mxnet.incubator.apache.org/api/python/autograd/autograd.html1.基本操作使用x.attach_grad()为梯度分配空间,调用with autograd.record()计算梯度,再使用backward()进行反传2. BN and dropoutBN, dropout这些层在训练和测试的时候是...原创 2019-03-18 10:04:22 · 2528 阅读 · 1 评论 -
mxnet学习(4): recordio
reference:https://mxnet.incubator.apache.org/api/python/io/io.html#mxnet.recordio.MXRecordIO1.class mxnet.recordio.MXRecordIO(uri, flag)读写RecordIO数据格式,支持顺序读写>>> record = mx.recordio.MXReco...原创 2019-03-18 10:09:08 · 3789 阅读 · 1 评论 -
mxnet学习(6):使用GPU进行训练
1.使用as_in_context()ctx是需要载入的gpu/cpu# Load data and label is the right context data = data.as_in_context(ctx)label = label.as_in_context(ctx)2.使用copyto(ctx)将数据copy到ctxdata = data.copyto(ctx)l...原创 2019-03-26 15:04:16 · 3228 阅读 · 2 评论 -
mxnet学习(7):数据载入方式
1.使用ImageRecordIter读取recmxnet.io.ImageRecordIter(*args, **kwargs)该方式只能从rec文件读取batches,相比于定制化的输入方式,该方式不够灵活,但是速度很快。如果要读取原图,可以使用ImageItereg:data_iter = mx.io.ImageRecordIter( path_imgrec="./sample....原创 2019-03-28 17:38:04 · 1633 阅读 · 0 评论 -
mxnet学习(8):Trainer
class mxnet.gluon.Trainer(params, optimizer, optimizer_params = None, kvstore = ‘device’, compression_params = None, updata_on_kvstore = None)参考:http://mxnet.incubator.apache.org/versions/master/api/...原创 2019-05-07 16:12:38 · 1922 阅读 · 0 评论 -
mxnet学习(9):使用gluon接口读取symbol预训练模型finetune
使用symbol得到的模型或者gluon的hybridize之后的模型包括一个.json文件(网络结构)和.params文件(参数),gluon可以使用net = gluon.SymbolBlock.imports(json, ['data'], params, ctx)导入网络和参数,这样可以进行测试或者进一步训练。但是如果只需要使用模型的其中一部分,比如只需要conv层,去掉所有fc层,或者...原创 2019-05-07 16:48:22 · 2179 阅读 · 1 评论