
472 Глава 14
>>> for j in range(num_epochs):
... img_batch, label_batch = next(iter(data_loader))
... img = img_batch[0]
... ax = fig.add_subplot(2, 5, j + 1)
... ax.set_xticks([])
... ax.set_yticks([])
... ax.set_title(f'Эпоха {j}:', size=15)
... ax.imshow(img.permute(1, 2, 0))
...
... img = img_batch[1]
... ax = fig.add_subplot(2, 5, j + 6)
... ax.set_xticks([])
... ax.set_yticks([])
... ax.imshow(img.permute(1, 2, 0))
>>> plt.show()
На рис. 14.17 показаны пять результатов преобразований для дополнения данных, при-
мененных к двум примерам изображений.
Рис. 14.17. Результаты пяти преобразований изображения
Далее мы применим функцию
transfor ...