三:运行预测
文件img_preprocess.py
包含了图像的预处理。
文件infer_resnet.py
包含了创建predictor,读取示例图片,预测,获取输出的等功能。
运行:
python infer_resnet.py --model_file=./ResNet50/model --params_file=./ResNet50/params --use_gpu=1
运行的结果为: (‘class index: ‘, 13)。
13表示图片的类别。我们通过imagenet 类别映射表, 可以找到对应的类别,即junco, snowbird,由此说明我们的分类器分类正确。