python-tensorflow和pytorch版本的手写数字识别

直接上图

选择图片，打开本地图片，进行识别，这个识别训练的是mnist数据集，所以这个白色区域内的数字会先经过黑白像素转后，传入训练好的模型中识别，识别结果在右侧显示。

训练的神经网络结构代码如下：

class LeNet(nn.Module):
    def __init__(self, num_classes=10):
        super(LeNet, self).__init__()
        self.conv1 = nn.Sequential(  # input_size=(1*28*28)
            nn.Conv2d(1, 6, 5, 1, 2),  # padding=2保证输入输出尺寸相同
            nn.ReLU(),  # input_size=(6*28*28)
            nn.MaxPool2d(kernel_size=2, stride=2),  # output_size=(6*14*14)
        )
        self.conv2 = nn.Sequential(
            nn.Conv2d(6, 16, 5),
            nn.ReLU(),  # input_size=(16*10*10)
            nn.MaxPool2d(2, 2)  # output_size=(1