From 05e3e11ebce2cd73ecdac20753ee7b70da322b71 Mon Sep 17 00:00:00 2001 From: yuejich Date: Sun, 16 May 2021 14:47:07 +0800 Subject: [PATCH] =?UTF-8?q?Add=20new=20file=20=E9=A2=84=E6=B5=8B-=E5=9B=BE?= =?UTF-8?q?=E7=89=87-=E4=B8=BB=E7=A8=8B=E5=BA=8F=20=E8=BF=90=E8=A1=8C?= =?UTF-8?q?=E8=BF=99=E4=B8=AA=E6=96=87=E4=BB=B6=EF=BC=8C=E4=BC=9A=E8=B0=83?= =?UTF-8?q?=E7=94=A8cnn=E5=8D=B7=E7=A7=AF=E7=A5=9E=E7=BB=8F=E7=BD=91?= =?UTF-8?q?=E7=BB=9C=EF=BC=8C=E5=8A=A0=E8=BD=BD=E4=B8=8Atrain=5Fmain.py?= =?UTF-8?q?=E8=AE=AD=E7=BB=83=E5=87=BA=E7=9A=84=E6=9D=83=E9=87=8D=E6=96=87?= =?UTF-8?q?=E4=BB=B6mnist.pdparams,=E9=A2=84=E6=B5=8B=E9=9C=80=E8=A6=81?= =?UTF-8?q?=E6=A3=80=E6=B5=8B=E7=9A=84=E5=9B=BE=E7=89=87=E4=B8=AD=E7=9A=84?= =?UTF-8?q?=E6=95=B0=E5=AD=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- predict_images_main.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 predict_images_main.py diff --git a/predict_images_main.py b/predict_images_main.py new file mode 100644 index 0000000..051770c --- /dev/null +++ b/predict_images_main.py @@ -0,0 +1,34 @@ +# 构建预测动态图过程 +import numpy as np +import paddle.fluid as fluid +from cnn_net import ConvolutionalNeuralNetwork +from images_chuli_function import load_image +from PIL import Image +import matplotlib.pyplot as plt + +with fluid.dygraph.guard(): + #新建模型 + model=ConvolutionalNeuralNetwork('mnist') + # 读取模型 + # 参数为保存模型参数的文件地址 + model_dict, _ = fluid.load_dygraph('mnist') + # 加载模型参数 + model.load_dict(model_dict) + #评估模式 + model.eval() + infer_path_3='infer_3.png' + infer_path_4='A4.jpg' + + ii=Image.open(infer_path_3) + plt.imshow(ii) + ii.show() + + img = load_image(infer_path_3) + + img=img/255 + + print(img.shape) + # 将np数组转换为dygraph动态图的variable + img = fluid.dygraph.to_variable(img) + result = model(img) + print('预测的结果是:{}'.format(np.argmax(result.numpy()))) -- GitLab