Pytorch Imag eNet набор данных - PullRequest
Новая
       66

Pytorch Imag eNet набор данных

1 голос
/ 09 марта 2020

Я не могу загрузить исходный набор данных Imag eNet с их официального сайта. Тем не менее, я обнаружил, что у pytorch есть Imag eNet в качестве одного из его наборов данных видения факела.

Q1. Это оригинальный набор данных Imag eNet?

Q2. Как получить классы для набора данных, как это делается в Cifar-10 

classes = [‘airplane’, ‘automobile’, ‘bird’, ‘cat’, ‘deer’, ‘dog’, ‘frog’, ‘horse’, ‘ship’, ‘truck’]

1 Ответ

0 голосов
/ 10 марта 2020

torchvision.datasets.ImageNet - это просто класс, который позволяет работать с набором данных Imag eNet, он не содержит изображений и меток Imag eNet сам по себе. Сначала необходимо загрузить набор данных Imag eNet, а затем его путь должен быть передан аргументу root torchvision.datasets.ImageNet. На данный момент вы должны загрузить его вручную с http://image-net.org/download-images (обратите внимание, что он имеет огромный размер, ~ 148 ГБ, поэтому, возможно, загрузка через торрент может быть более подходящим вариантом, например, поиск по академическим скидкам - но Я не уверен, что это законно), так как он больше не доступен для общественности: 

if download is True:
    msg = ("The dataset is no longer publicly accessible. You need to "
           "download the archives externally and place them in the root "
           "directory.")
    raise RuntimeError(msg)
elif download is False:
    msg = ("The use of the download flag is deprecated, since the dataset "
           "is no longer publicly accessible.")
    warnings.warn(msg, RuntimeWarning)

( source )

Если вам просто нужно получить имена классов и соответствующие индексы без загрузки всего набора данных (например, если вы используете предварительно обученную модель и хотите отобразить прогнозы на метки), вы можете загрузить их, например, с здесь или с этого Github Gist.

...