优草派  >   Python

Python深度神经网络入门怎么准备数据?PyTorch模型训练数据准备代码

罗西汉            来源:优草派

python是人工智能重要支撑开发语言之一,而人工智能是由很多不同领域来组成的,其中一个就是深度神经网络学习。而它是使用大量数据训练来构建模型的,所以下文所主要介绍的内容就是pytorch深度神经网络入门,模型训练准备数据的代码,往下看看吧。

Python深度神经网络入门怎么准备数据?PyTorch模型训练数据准备代码

1.pytorch基本上是现在所有python神经网络学习程序都在使用的第三方库和框架,而它有一个很强大的功能就是,可以来对大量图片数据进行训练从而得到图像特征模型。那么这种方式需要先将准备好的图片数据放在一个文件夹之中,接下来就是通过模型训练准备代码来将其重命名并且生成标签文本文档了。

2.打开pycharm或者是spyder这些专业开发工具,然后再去新建python文件作为脚本编写代码,整体diam的详细示例如下所示:

import torch
import torchvision
import matplotlib.pyplot as plt
from skimage import io
mnist_test= torchvision.datasets.MNIST(
    './mnist', train=False, download=True
)
f=open('mnist_test.txt','w')
for i,(img,label) in enumerate(mnist_test):
    img_path="./mnist_test/"+str(i)+".jpg"
    io.imsave(img_path,img)
    f.write(img_path+' '+str(label)+'\n')
f.close()

上面代码之中使用神经网络学习库pytorch之中的数据集处理方法Mnist访问了一个文件路径,而这个文件路径就是保存了图片数据的文件夹,之后会返回一个数据集对象。然后使用open()函数新建txt文档,在for循环之中将数据集全部图片名称取出来并且重命名和加上标签,这样在每一个图片都会被按照顺序命名并且对其按照标签进行分组。

以上就是关于“Python深度神经网络入门怎么准备数据?PyTorch模型训练数据准备代码”的全部内容了,希望对你有所帮助。

【原创声明】凡注明“来源:优草派”的文章,系本站原创,任何单位或个人未经本站书面授权不得转载、链接、转贴或以其他方式复制发表。否则,本站将依法追究其法律责任。
TOP 10
  • 周排行
  • 月排行