Pytorch学习（二）—— Dataset类

上文我们已经配置好环境，接下来进入实战环节。

Dataset类实战

查看image的各种属性

首先在pycharm中创建py文件。

从torch中导入Dataset。

因为我们要对图像进行处理，我们再导入PIL（Python Imaging Library）

查看文件路径，导入os

在python console（容易查看各种属性）中重复上述操作：

查看一张照片：

右键照片位置-复制路径-选择absolute（绝对路径）

用img_path接收，用img读取。

（注意是双\）

console的右边就有信息：

打开img有更详细的信息，在console中用img.xx查看。

eg.

创建一个子类，继承父类Dataset

重写父类方法，init和getitem。

想通过idx返回图片地址的话，首先创建图片列表

这下就可以在getitem中实现对图片的读取了。

再写一下返回len：

创建两个对象ants和bees，二者可以用"+"创建集合。

这个对于后续想扩展数据集结合训练很有用。