pytorch训练自己的数据集

将图片放在每类的文件夹下面，接着放一个txt文件 使用img_data = torchvision.datasets.ImageFolder(image_path, transform)