Загрузите предварительно обученную сеть генератора UNIT с помощью функции helper 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, которая переводит изображения между дневным и сумеречным условиями освещения с помощью функции helper 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)