From dbe806b8e80f767ce7fb70faacb96b75211d2059 Mon Sep 17 00:00:00 2001 From: Dragon Fire Date: Thu, 9 May 2024 15:31:23 -0400 Subject: [PATCH] Fix --- structures/Tensorflow.js | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/structures/Tensorflow.js b/structures/Tensorflow.js index a0d04ca2..47238b24 100644 --- a/structures/Tensorflow.js +++ b/structures/Tensorflow.js @@ -74,7 +74,7 @@ module.exports = class Tensorflow { async stylizeImage(image, styleImg) { const imageTensor = await tf.node.decodeImage(image, 3); const [originalHeight, originalWidth] = imageTensor.shape.slice(0, 2); - const desiredHeight = 400; + const desiredHeight = 256; const aspectRatio = originalWidth / originalHeight; const newWidth = Math.round(desiredHeight * aspectRatio); const resizedImage = tf.image.resizeBilinear(imageTensor, [desiredHeight, newWidth]);