在训练得到一个模型后,一般可以通过一个测试代码测试模型在指定输入图像时的输出结果,从而判断训练得到的模型是否有效。有时候测试代码也被称为demo,可以用来展示模型训练的效果,因此测试代码是一个完整项目必不可少的部分。
完整的测试代码如代码清单4-2所示,主要包含模型导入、数据读取和预测输出3个部分,下面假设这些代码都保存在test_mnist_code4-2.py脚本中。
import mxnet as mx
import numpy as np
def load_model(model_prefix, index, context, data_shapes, label_shapes):
sym, arg_params