💥今天看一下 PyTorch数据通常的处理方法~
一般我们会将dataset用来封装自己的数据集,dataloader用于读取数据
💬dataset定义了这个数据集的总长度,以及会返回哪些参数,模板:
💬输出:
💬如果修改batch_size为2,则输出:
💥dataset只是一个类,因此数据可以从外部导入,我们也可以在dataset中规定数据在返回时进行更多的操作,数据在返回时也不一定是有两个。
需要安装opencv
pip install opencv-python
有一些数据集是PyTorch自带的,它被保存在TorchVision
中,以mnist
数据集为例进行加载: