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]);