Блог

Я реализовал UNet в PyTorch lightning, и я получаю результаты. Теперь я хотел бы сохранить все выходные данные тестов в папку с тем же именем, что и у тестовых изображений. Например, если имя тестового изображения-image1, то мне нужно сохранить тестовое выходное изображение также как image1. Я просмотрел документацию по молнии PyTorch, о которой мне стало известно

тренер.

Но я не знаю, как это реализовать.