nsfwjs-node/read.js

Detect NSFW content client side

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var tf = require("@tensorflow/tfjs-node");
var fs = require('fs');
var jpeg = require('jpeg-js');
exports.NUMBER_OF_CHANNELS = 3;
exports.readImage = function (path) {
    var buf = fs.readFileSync(path);
    return jpeg.decode(buf, {
        useTArray: true
    });
};
exports.imageByteArray = function (image, numChannels) {
    var pixels = image.data;
    var numPixels = image.width * image.height;
    var values = new Int32Array(numPixels * numChannels);
    for (var i = 0; i < numPixels; i++) {
        for (var channel = 0; channel < numChannels; ++channel) {
            values[i * numChannels + channel] = pixels[i * 4 + channel];
        }
    }
    return values;
};
exports.imageToInput = function (image, numChannels) {
    var values = exports.imageByteArray(image, numChannels);
    var outShape = [image.height, image.width, numChannels];
    return tf.tensor3d(values, outShape, 'int32');
};
//# sourceMappingURL=read.js.map