Как из float32 изображения сделать jpg?

Question

[Ася]

У меня есть одна картинка которую я добавляю так:

let cat1 = new Image;
    cat1.src = "cat-1.jpg";

И второе изображение:

let cat2 = tf.squeeze(result); 
console.log(cat2.dtype) #float32

И мне нужно, чтобы переменная cat2 была аналогична cat1.

Comments

[Сергей Соколов]

а что там вообще лежит в cat2 после sqeeze() ? Какой shape? Какой диапазон значений: 0..1 ?

[Ася]

shape: 300, 300, 3
range: [0-1]

Answer 1

[Сергей Соколов]

Наивный ответ без учёта всех возможных косяков из-за неполноты вопроса:

const tensor = tf.squeeze(result);

const canvas = document.createElement('canvas');
canvas.width = tensor.shape.width
canvas.height = tensor.shape.height
await tf.browser.toPixels(tensor, canvas);

const cat2 = document.createElement('img');
cat2.src = canvas.toDataURL();

Comments

[Ася]

почему-то создается пустой канвас (в нем не отображается изображение)
тензор у меня такой:

[Сергей Соколов]

Ася, вроде всё в порядке: когда тип float32, ожидаются значения 0..1 для отрисовки. Если бы тип был int, то 0..255

Может, значения в тензоре все по нулям, в самом деле?
await не потеряли? toPixels() возвращает Promise, который ресолвится не мгновенно.