qq_39663923的博客

原创 基于pytorch，实现手写数字识别

这里可以看到我们在定义一个Mydata类时要传入两个变量一个是包好数据和标签的txt文件，另一个是device类型的参数。为使方法具有通用性，这里将张量转化为图片并以 data -> train/val -> 类别1/2/... -> 图片的形式进行存储。因为模型的输入为单通道28*28，所以使用前要将目的图形转化为灰度图，并且分辨率改为28*28.然后我们提取各个图片的位置，并创建两个字典，以 位置：标签的形式保存到字典当中。数据处理的方法和类完成后，我们接下来进行模型的搭建。