PyTorch: Testing with torchvision.datasets.ImageFolder and DataLoader

后端 未结 2 1689
礼貌的吻别
礼貌的吻别 2021-01-11 15:42

I\'m a newbie trying to make this PyTorch CNN work with the Cats&Dogs dataset from kaggle. As there are no targets for the test images, I manually classified some of the

2条回答
  •  生来不讨喜
    2021-01-11 16:15

    As per @Monster's comment above here is my folder structure for ImageFolder

    And this how I load the dataset:

        train_dataset=datasets.ImageFolder(root="./root/",transform=train_transforms)
    

提交回复
热议问题