mxnet 入门0

初始化

 ####### 如果有预训练模型的话
sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, epoch)  ######### prefix 后会自动搜索-symbol.json的后缀

######### 中间层作为输出
    all_layers = sym.get_internals()
    sym = all_layers['fc1_output']
self.image_size = image_size

######### 重建模型
    model = mx.mod.Module(symbol=sym, context=ctx, label_names = None)
    model.bind(for_training=False, data_shapes=[('data', (1, 3, image_size[0], image_size[1]))])
    model.set_params(arg_params, aux_params)

运行

    input_blob[0] = img
    data = mx.nd.array(input_blob)
    db = mx.io.DataBatch(data=(data,))
    self.model.forward(db, is_train=False)
    pred = self.model.get_outputs()[-1].asnumpy()[0