将图片放在每类的文件夹下面,接着放一个txt文件
使用img_data = torchvision.datasets.ImageFolder(image_path, transform)
pytorch训练自己的数据集
最新推荐文章于 2024-07-09 08:56:16 发布
将图片放在每类的文件夹下面,接着放一个txt文件
使用img_data = torchvision.datasets.ImageFolder(image_path, transform)