Fix

lilyissillyyy · May 9, 2024 · f969002 · f969002
1 parent f2147f3
commit f969002
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/structures/Tensorflow.js b/structures/Tensorflow.js
@@ -77,7 +77,7 @@ module.exports = class Tensorflow {
 		const desiredHeight = 400;
 		const aspectRatio = originalWidth / originalHeight;
 		const newWidth = Math.round(desiredHeight / aspectRatio);
-		const resizedImage = tf.image.resizeBilinear(imageTensor, [desiredHeight, newWidth]);
+		const resizedImage = tf.image.resizeBilinear(imageTensor, [newWidth, desiredHeight]);
 		imageTensor.dispose();
 		const loadedImage = resizedImage.toFloat().div(tf.scalar(255)).expandDims();
 		resizedImage.dispose();