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

Fix

Browse Source
This commit is contained in:
Dragon Fire
2024-05-09 15:09:52 -04:00
parent cd4dc65420
commit 458f545ff5
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
structures/Tensorflow.js
+7 -1
View File
@@ -73,8 +73,14 @@ 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 loadedImage = imageTensor.toFloat().div(tf.scalar(255)).expandDims(); const [originalHeight, originalWidth] = imageTensor.shape.slice(0, 2);
const desiredWidth = 400;
const aspectRatio = originalWidth / originalHeight;
const newHeight = Math.round(desiredWidth / aspectRatio);
const resizedImage = tf.image.resizeBilinear(imageTensor, [newHeight, desiredWidth]);
imageTensor.dispose(); imageTensor.dispose();
const loadedImage = resizedImage.toFloat().div(tf.scalar(255)).expandDims();
resizedImage.dispose();
const styleTensor = tf.node.decodeImage(styleImg, 3); const styleTensor = tf.node.decodeImage(styleImg, 3);
const loadedStyle = styleTensor.toFloat().div(tf.scalar(255)).expandDims(); const loadedStyle = styleTensor.toFloat().div(tf.scalar(255)).expandDims();
styleTensor.dispose(); styleTensor.dispose();