码迷,mamicode.com
首页 > 编程语言 > 详细

Python 10 训练模型

时间:2019-07-19 10:58:05      阅读:211      评论:0      收藏:0      [点我收藏+]

标签:span   div   完成   其它   ORC   log   zip   png   nis   

原文:https://www.cnblogs.com/denny402/p/7520063.html

原文:https://www.jianshu.com/p/84f72791806f

 

 

1、下载mnist数据集

  地址:http://yann.lecun.com/exdb/mnist/

  下面这四个都要下载,下载完成后,解压到同一个目录,我是解压到“E:/fashion_mnist/”这个目录里面,好和下面的代码目录一致

  解压完成后,需要修改一下文件名,如(修改原因:保持和下面代码一样,避免出现其它问题):

    修改前:t10k-images.idx3-ubyte

    修改后:t10k-images-idx3-ubyte

  我是第一次弄这玩意,所以尽量弄得白痴些,走弯路很烦,有时候一点点小问题就弄半天,其实就是别人有那么一点没讲清楚,然后就会搞很久

技术图片

技术图片

 

 

 

 

2、执行原文1里面的这段代码。

  这段代码里面,需要先用pip安装skimage、torch、torchvision,前两篇文章有安装步骤。

import os
from skimage import io
import torchvision.datasets.mnist as mnist

root="E:/fashion_mnist/"
train_set = (
    mnist.read_image_file(os.path.join(root, train-images-idx3-ubyte)),
    mnist.read_label_file(os.path.join(root, train-labels-idx1-ubyte))
        )
test_set = (
    mnist.read_image_file(os.path.join(root, t10k-images-idx3-ubyte)),
    mnist.read_label_file(os.path.join(root, t10k-labels-idx1-ubyte))
        )
print("training set :",train_set[0].size())
print("test set :",test_set[0].size())

def convert_to_img(train=True):
    if(train):
        f=open(root+train.txt,w)
        data_path=root+/train/
        if(not os.path.exists(data_path)):
            os.makedirs(data_path)
        for i, (img,label) in enumerate(zip(train_set[0],train_set[1])):
            img_path=data_path+str(i)+.jpg
            io.imsave(img_path,img.numpy())
            f.write(img_path+ +str(label)+\n)
        f.close()
    else:
        f = open(root + test.txt, w)
        data_path = root + /test/
        if (not os.path.exists(data_path)):
            os.makedirs(data_path)
        for i, (img,label) in enumerate(zip(test_set[0],test_set[1])):
            img_path = data_path+ str(i) + .jpg
            io.imsave(img_path, img.numpy())
            f.write(img_path +   + str(label) + \n)
        f.close()

convert_to_img(True)
convert_to_img(False)

 

Python 10 训练模型

标签:span   div   完成   其它   ORC   log   zip   png   nis   

原文地址:https://www.cnblogs.com/guxingy/p/11211714.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!