unitPredict

Выполните вывод с помощью безнадзорной сети (UNIT) перевода от изображения к изображению

Описание

пример

translatedImage = unitPredict(net,inputImage) выполняет безнадзорный перевод от изображения к изображению изображения inputImage использование сети UNIT net.

Эта функция требует Deep Learning Toolbox™.

пример

translatedImage = unitPredict(net,inputImage,"OutputType",outputType) задает направление перевода от изображения к изображению для вывода с помощью outputType аргумент. Направление может быть источником к цели или целью к источнику.

Примеры

свернуть все

Загрузите предварительно обученную МОДУЛЬНУЮ сеть генератора, которая переводит изображения между дневным временем и условиями освещения сумрака с помощью функции помощника 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");

Переведите исходное изображение в целевую область с помощью МОДУЛЬНОЙ сети генератора.

dlOutput = unitPredict(gen,dlInput);

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

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

Загрузите предварительно обученную МОДУЛЬНУЮ сеть генератора, которая переводит изображения между дневным временем и условиями освещения сумрака с помощью функции помощника 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");

Переведите целевое изображение в исходную область использование предварительно обученной МОДУЛЬНОЙ сети генератора, gen.

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

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

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

Входные параметры

свернуть все

МОДУЛЬНАЯ сеть генератора в виде dlnetwork Объект (Deep Learning Toolbox). Можно создать МОДУЛЬНУЮ сеть генератора использование unitGenerator функция.

Введите изображение для перевода от изображения к изображению в виде отформатированного dlarray Объект (Deep Learning Toolbox).

Направление перевода от изображения к изображению для вывода в виде одного из этих значений.

  • "SourceToTarget" – переведите от исходной области до целевой области

  • "TargetToSource" – переведите от целевой области до исходной области

Типы данных: char | string

Выходные аргументы

свернуть все

Выведенное изображение после перевода от изображения к изображению, возвращенного как dlarray Объект (Deep Learning Toolbox).

Введенный в R2021a