PyTorch & Google Colab: ImageFolder загружает не все изображения - PullRequest
0 голосов
/ 07 февраля 2019

Я использую PyTorch для задачи классификации изображений «Кошки против собак» (9999 изображений кошек и 9999 изображений собак) в Google colab со следующим фрагментом кода для создания загрузчика данных:

import torchvision
from torchvision import transforms
data = torchvision.datasets.ImageFolder(r'/content/drive/My Drive/kaggle/trainset/trainset',transform = transforms.ToTensor())

и «data» имеют длину 6594 (против 19998 ожидаемых).Я пытаюсь перезапустить среду выполнения и каждый раз получаю разное количество изображений.Затем я попытался запустить тот же кусок кода в блокноте Jupyter, и он работал как положено, загрузив все 19998 изображений.Что я делаю не так в колабе?

...