pytorch自定义数据集

使用函数torch.utils.data.Datasettorch.utils.data.DataLoader数据准备以猫狗为例实现分类按照如下图所示建立文件和文件夹，我这里自己准备了20张猫狗图像。test.txt文件是后面代码生成的，先不用管，cats和dogs里面放上自己的图片，然后通过脚本生成test.txt文件，text.txt的脚本 代码如下：#!/usr/bin/python# -*- coding:utf-8 -*-import osdef generate(dir,