猫狗数据集的分为训练集25000张,在训练集中猫和狗的图像是混在一起的,pytorch读取数据集有两种方式,第一种方式是将不同类别的图片放于其对应的类文件夹中,另一种是实现读取数据集类,该类继承torch.utils.Dataset,并重写__getitem__和__len__. 先将猫和狗从训练集中区分开来,分别放到dog和cat文件夹下: import glob import shutil import os #数据集目录 path = "./ml/dogs-vs-cats/train&qu