MINIST数据集手写数字识别单张图片读入测试代码

import torch
import torch.nn as nn
import torchvision
from #所在程序名字 import #自己网络保存的名字
import matplotlib.pyplot as plt
from PIL import Image
from torchvision import transforms
import numpy as npdef test_mydata():im = plt.imread('33333.jpg')     #自己手写的图片读入    images = Image.open('33333.jpg')  #可以转换为黑底白字，读取更准确images = images.resize((28,28))images = images.convert('L')transform = transforms.ToTensor()images = transform(images)images = images.resize(1,1,28,28)# 加载网络和参数model = #加上自己网络的名字()model.load_state_dict(torch.load('pathh'))model.eval()outputs = model(images)values, indices = outputs.data.max(1) # 返回最大概率值和下标plt.title('{}'.format((int(indices[0]))))plt.imshow(im)plt.show()test_mydata()

MINIST数据集手写数字识别单张图片读入测试代码相关推荐

[Pytorch系列-41]：卷积神经网络 - 模型参数的恢复/加载 - 搭建LeNet-5网络与MNIST数据集手写数字识别
作者主页(文火冰糖的硅基工坊):文火冰糖(王文兵)的博客_文火冰糖的硅基工坊_CSDN博客本文网址:https://blog.csdn.net/HiWangWenBing/article/detai ...
pyTorch入门（六）——实战Android Minist OpenCV手写数字识别（附源码地址）
学更好的别人, 做更好的自己. --<微卡智享> 本文长度为4239字,预计阅读12分钟前言前面几篇文章实现了pyTorch训练模型,然后在Windows平台用C++ OpenCV D ...
MNIST数据集手写数字识别
1 数据集介绍 MNIST 包括6万张28x28的训练样本,1万张测试样本,很多教程都会对它"下手"几乎成为一个 "典范",可以说它就是计算机视觉里面的Hell ...
【项目实践】：KNN实现手写数字识别（附Python详细代码及注释）
↑ 点击上方[计算机视觉联盟]关注我们本节使用KNN算法实现手写数字识别.KNN算法基本原理前边文章已经详细叙述,盟友们可以参考哦! 数据集介绍有两个文件: (1)trainingDigits文件 ...
pytorch MNIST 手写数字识别 + 使用自己的测试集 + 数据增强后再训练
文章目录 1. MNIST 手写数字识别 2. 聚焦数据集扩充后的模型训练 3. pytorch 手写数字识别基本实现 3.1完整代码及 MNIST 测试集测试结果 3.1.1代码 3.1.2 MNI ...
卷积神经网络手写数字识别（包含Pytorch实现代码）
Hello!欢迎来到六个核桃Lu! 运用卷积神经网络实现手写数字识别 1 算法分析及设计卷积神经网络: 图1-2 如图1-2,卷积神经网络由若干个方块盒子构成,盒子从左到右仿佛越来越小,但却越来越 ...
Python深度学习之分类模型示例，MNIST数据集手写数字识别
MNIST数据集是机器学习领域中非常经典的一个数据集,由60000个训练样本和10000个测试样本组成,每个样本都是一张28 * 28像素的灰度手写数字图片. 我们把60000个训练样本分成两部分,前 ...
MNIST数据集手写数字识别（一）
MNIST数据集是初步学习神经网络的很好的数据集,也是拿来教学,不可多得的好教材,有很多知识点在里面.官网下载地址,可以自己手动下载,当然也可以通过下面的代码自动下载[urllib.request(3 ...
MNIST数据集手写数字识别（二）
上一篇对MNIST数据集有了一些了解,数据集包含着60000张训练图片与标签值和10000张测试图片与标签值的数据集,数据集有了,现在我们来构造神经网络,预测下对这测试的10000张图片的正确识别率, ...
linux手写数字识别,mnist手写数字识别与图片预处理
用户mnist训练了一个模型,想用自己的图片进行测试,需要将自己的图片进行预处理为mnist需要的1*784的二维张量. 因为MNIST数据是28*28的黑底白字图像,而且输入时要将其拉直,也就是可以 ...

MINIST数据集手写数字识别单张图片读入测试代码

MINIST数据集手写数字识别单张图片读入测试代码相关推荐

最新文章

热门文章