Загрузка предварительно обученной сети генератора UNIT с помощью функции помощника downloadTrainedDayDuskGeneratorNet. Генераторная сеть переводит изображения между дневными и сумерками. Исходная область - дневное освещение, а целевая область - сумерки.

trainedUNIT_url = 'https://ssd.mathworks.com/supportfiles/vision/data/trainedDayDuskUNITGeneratorNet.zip';
trainedUNIT_filename = 'trainedDayDuskUNITGeneratorNet.mat';
downloadTrainedDayDuskGeneratorNet(trainedUNIT_url,pwd);
load(trainedUNIT_filename);

Считывание и отображение тестового изображения, полученного в дневное время.

input = imread("car1.jpg");
imshow(input)

Выполните предварительную обработку образа, чтобы он был совместим с сетью. Преобразование данных в тип данных single в диапазоне [-1, 1]. Уменьшите размер изображения и сохраните данные в dlarray объект.

input = (im2single(input) - 0.5)/0.5;
input = imresize(input,0.25);
dlInput = dlarray(input,"SSCB");

Преобразование исходного изображения в целевую область с помощью предварительно обученной сети генератора UNIT. gen.

dlOutput = unitPredict(gen,dlInput);

Извлеките преобразованные данные изображения из dlarray и масштабировать данные до диапазона [0, 1]. Отображение переведенного изображения. Переведенное изображение напоминает изображения, полученные в сумерках.

output = rescale(extractdata(dlOutput));
imshow(output)

Загрузите предварительно обученную сеть генератора UNIT, которая переводит изображения между дневными и сумерками освещения с помощью функции помощника downloadTrainedDayDuskGeneratorNet. Исходная область - дневное освещение, а целевая область - сумерки.

trainedUNIT_url = 'https://ssd.mathworks.com/supportfiles/vision/data/trainedDayDuskUNITGeneratorNet.zip';
trainedUNIT_filename = 'trainedDayDuskUNITGeneratorNet.mat';
downloadTrainedDayDuskGeneratorNet(trainedUNIT_url,pwd);
load(trainedUNIT_filename);

Считывание и отображение тестового изображения, полученного в сумерках.

input = imread("office_2.jpg");
imshow(input)

Выполните предварительную обработку образа, чтобы он был совместим с сетью. Преобразование данных в тип данных single в диапазоне [-1, 1]. Сохранение данных в dlarray объект.

input = (im2single(input) - 0.5)/0.5;
dlInput = dlarray(input,"SSCB");

Преобразуйте целевой образ в исходный домен с помощью предварительно обученной сети генератора UNIT, gen.

dlOutput = unitPredict(gen,dlInput,"OutputType","TargetToSource");

Извлеките преобразованные данные изображения из dlarray и масштабировать данные до диапазона [0, 1]. Отображение переведенного изображения. Переведенное изображение напоминает изображения, полученные в дневных условиях освещения.

output = rescale(extractdata(dlOutput));
imshow(output)