arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-03 23:36:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Dragon Fire
2024-05-09 15:31:23 -04:00
parent 952e0f8d22
commit dbe806b8e8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
structures/Tensorflow.js
+1 -1
View File
@@ -74,7 +74,7 @@ module.exports = class Tensorflow {
async stylizeImage(image, styleImg) { async stylizeImage(image, styleImg) {
const imageTensor = await tf.node.decodeImage(image, 3); const imageTensor = await tf.node.decodeImage(image, 3);
const [originalHeight, originalWidth] = imageTensor.shape.slice(0, 2); const [originalHeight, originalWidth] = imageTensor.shape.slice(0, 2);
const desiredHeight = 400; const desiredHeight = 256;
const aspectRatio = originalWidth / originalHeight; const aspectRatio = originalWidth / originalHeight;
const newWidth = Math.round(desiredHeight * aspectRatio); const newWidth = Math.round(desiredHeight * aspectRatio);
const resizedImage = tf.image.resizeBilinear(imageTensor, [desiredHeight, newWidth]); const resizedImage = tf.image.resizeBilinear(imageTensor, [desiredHeight, newWidth]);