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

Fix

Browse Source
This commit is contained in:
Dragon Fire
2024-05-08 18:30:25 -04:00
parent be7ef70087
commit dc12937bbc
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
structures/Tensorflow.js
+1
View File
@@ -83,6 +83,7 @@ module.exports = class Tensorflow {
const stylizedImage = await this.transformerModel.predict([loadedImage, stylePrediction]); const stylizedImage = await this.transformerModel.predict([loadedImage, stylePrediction]);
loadedImage.dispose(); loadedImage.dispose();
stylePrediction.dispose(); stylePrediction.dispose();
stylizedImage.mul(255).toInt();
const buffer = await tf.node.encodeJpeg(stylizedImage.squeeze()); const buffer = await tf.node.encodeJpeg(stylizedImage.squeeze());
stylizedImage.dispose(); stylizedImage.dispose();
return Buffer.from(buffer); return Buffer.from(buffer);