
Глубокое обучение нейронных сетей на основе PyTorch
379
>>> image_dataset = ImageDataset(file_list, labels)
>>> for file, label in image_dataset:
... print(file, label)
cat_dog_images/cat-01.jpg 0
cat_dog_images/cat-02.jpg 0
cat_dog_images/cat-03.jpg 0
cat_dog_images/dog-01.jpg 1
cat_dog_images/dog-02.jpg 1
cat_dog_images/dog-03.jpg 1
Объединенный набор данных содержит имена файлов и метки.
Затем к этому набору данных нужно применить следующие операции: загрузить со-
держимое изображения по его пути к файлу, декодировать необработанное содержимое
и изменить его размер до желаемого значения — например, 80×120. Воспользуемся ...