在学习pytorch深度学习的过程中,通常会使用到大量的数据集,包括训练集和测试集。

以下是pytorch加载数据集的流程:

在notebook中使用help方法查看Dataset类的功能以及操作:

使用dataset需要继承Dataset父类 重写__getitem__方法和__len__方法 __getitem__():由给定的key获取数据集中每个图片的操作函数 __len__():获取数据集中图片大小的函数

以上是Dataset类的功能以及使用方法。

下载数据集:链接: https://pan.baidu.com/s/1jZoTmoFzaTLWh4lKBHVbEA 密码: 5suq

将下载好的数据集解压到数据集文件夹命名dataset

在pycharm中输入代码:

安装Pillow库:Pillow是一个Python图像处理库,它提供了处理图像的各种功能,包括打开、保存、调整大小、裁剪、旋转等。在PyCharm中,可以使用以下命令安装Pillow库:

pip install Pillow

导入Image模块:在Python代码中,首先需要导入Image模块。可以使用以下语句导入Image模块:

from PIL import Image

打开图像文件:使用Image模块的open()函数可以打开图像文件。可以通过指定图像文件的路径来打开图像。例如,如果图像文件位于与你的Python文件相同的目录中,你可以使用以下代码打开图像文件:

image = Image.open("image.jpg")

进行图像处理操作:一旦打开了图像文件,就可以使用Image对象上的各种方法来进行图像处理操作。例如,可以调整图像的大小、裁剪图像、旋转图像等。以下是一些常用的图像处理操作示例:

# 调整图像大小

resized_image = image.resize((width, height))

# 裁剪图像

cropped_image = image.crop((left, top, right, bottom))

# 旋转图像

rotated_image = image.rotate(angle)

保存图像:完成图像处理后,可以使用Image对象的save()方法将图像保存到文件中。可以指定保存的文件路径和文件格式。例如,可以使用以下代码将图像保存为JPEG格式:

resized_image.save("resized_image.jpg", "JPEG")

这些是使用Image包进行基本图像处理的一些步骤。可以根据自己的需求使用更多的Image模块方法来进行高级的图像处理操作。

实例操作:

输出结果:

参考链接

评论可见,请评论后查看内容,谢谢!!!评论后请刷新页面。