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
2021-03-22 18:10:01 -04:00
parent e4700aef19
commit de4ffb24a6
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
util/Util.js
+9 -2
View File
@@ -1,5 +1,6 @@
const crypto = require('crypto');
const tf = require('@tensorflow/tfjs-node');
const sizeOf = require('buffer-image-size');
const { decode: decodeHTML } = require('html-entities');
const { stripIndents } = require('common-tags');
const { URL } = require('url');
@@ -218,8 +219,14 @@ module.exports = class Util {
}
static async isImageNSFW(model, image, bool = true) {
const img = await tf.node.decodeImage(new Uint8Array(image), 3);
const predictions = model.classify(image, 1);
const dimensions = sizeOf(image);
const data = {
data: new Uint8Array(image),
width: dimensions.width,
height: dimensions.height
};
const img = await tf.node.decodeImage(data, 3);
const predictions = await model.classify(image, 1);
img.dispose();
return bool ? predictions[0] !== 'Neutral' && predictions[0] !== 'Drawing' : predictions[0];
}