aistudio作为百度旗下的人工智能训练平台,可以免费供各位深度学习爱好者使用训练模型,aistudio下内置paddle框架,支持动态图与静态图的训练。此外,aistudio还支持实现模型的在线部署,我们可以利用aistudio平台训练模型,然后在aistudio上进行部署,通过api的方式调用模型服务。在aistudio中部署模型的教程如下链接所示:https://ai.baidu.com/ai-doc/AISTUDIO/bk3e382cq。该网页主要讲述了静态图模型的部署,只涉及到数值数据的输入输出,不支持图像的输入,也不支持动态图的部署。为此,博主对官网教程进行拓展研究,最终实现动态图的部署测试,本博文以minist数据集分类模型为例进行讲解。
这里博主针对minisit数据集搭建了一个简单的分类网络并进行训练保存,如果各位有自己已经训练好的动态图模型则大可不必进行这个步骤。
import paddle
from paddle import nn
from paddle.vision.transforms import ToTensor