在我的自定义数据集中，一种图像在一个文件夹中，torchvision.datasets.Imagefolder 可以处理，但如何将数据集拆分为训练和测试？
machine-learning image-processing computer-vision pytorch
computer-vision ×1
image-processing ×1
machine-learning ×1
pytorch ×1