Increase tensorflow max size

This commit is contained in:
Dragon Fire
2024-05-05 20:51:47 -04:00
parent f5d393d58a
commit fcb5896b01
+1 -1
View File
@@ -29,7 +29,7 @@ module.exports = class Tensorflow {
} }
async detectFaces(imgData) { async detectFaces(imgData) {
if (Buffer.byteLength(imgData) >= 4e+6) return 'size'; if (Buffer.byteLength(imgData) >= 8e+6) return 'size';
tfnode.setBackend('tensorflow'); tfnode.setBackend('tensorflow');
const image = tfnode.node.decodeImage(imgData, 3); const image = tfnode.node.decodeImage(imgData, 3);
tfnode.setBackend('cpu'); tfnode.setBackend('cpu');