如何用PyTorch自定义DataSet实现图像数据集的加载与高级操作？

本文共计1786个文字，预计阅读时间需要8分钟。

使用PyTorch自定义DataSet，以加载图像数据集为例，实现一些操作+总共分为四步+构造一个my_dataset类+继承自torch.utils.data.Dataset+重写__getitem__和__len__方法

使用pytorch自定义DataSet，以加载图像数据集为例，实现一些骚操作

总共分为四步

• 构造一个​​my_dataset​​​类，继承自​​torch.utils.data.Dataset​​
• 重写​​__getitem__​​​ 和​​__len__​​ 类函数
• 建立两个函数​​find_classes​​​、​​has_file_allowed_extension​​，直接从这copy过去
• 建立​​my_make_dataset​​函数用来构造(path,lable)对

一、构造一个​​my_dataset​​​类，继承自​​torch.utils.data.Dataset​​

二、 重写​​__getitem__​​​ 和​​__len__​​ 类函数

要构造Dataset的子类，就必须要实现两个方法：

• getitem_(self, index)：根据index来返回数据集中标号为index的元素及其标签。