深度学习经典案例解析:基于MATLAB
上QQ阅读APP看书,第一时间看更新

3.2 实例需求与实现步骤

【例3-1】 构建并训练一个卷积神经网络,对输入图像中(像素为28×28)数字的倾斜角度进行预测,计算预测准确率和均方根误差(RMSE)。部分输入图像如图3-4所示。

图3-4 部分输入图像

上述需求可以通过以下5个步骤实现:

步骤1:加载图像样本数据。

步骤2:将加载的图像样本分为训练集和测试集。

步骤3:构建卷积神经网络。

步骤4:配置训练选项并开始训练。

步骤5:将训练好的网络用于对新的输入图像进行分类,并计算准确率和均方根误差。