paddlepaddle的手写体识别

最新推荐文章于 2025-08-30 01:41:55 发布

原创

最新推荐文章于 2025-08-30 01:41:55 发布 · 390 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#paddlepaddle #pytorch #深度学习

本文介绍了如何使用PaddlePaddle库构建一个简单的MNIST手写数字识别模型，包括数据预处理、训练过程和模型预测。从加载预训练模型到对新图片进行分类，展示了深度学习基础应用。

import paddle
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image
import paddle.nn.functional as F


class MNIST(paddle.nn.Layer):
    def __init__(self):
        super(MNIST, self).__init__()

        self.fc = paddle.nn.Linear(in_features=784, out_features=1)

    def forward(self, inputs):
        outputs = self.fc(inputs)
        return outputs


model = MNIST()

def norm_img(img):
    assert len(img.shape) == 3
    batch_size, img_h, img_w = img.shape[0], img.shape[1], img.shape[2]

    img = img / 255