index.js

const tf = require('@tensorflow/tfjs-node');
const nsfw = require('nsfwjs');

const express = require('express');
const app = express();
// const fs = require("fs");

const port = 3000;
const nsfwLoad = nsfw.load();

app.post('/nsfw', (req, res) => {
    try {
        let data = [];
        req.on("data", (chunk) => {
            try {
                data.push(chunk);
            } catch (e) {
                res.end(JSON.stringify({
                    'code': 500,
                    'msg': e.message,
                    'data': ''
                }));
            };
        });
        req.on("end", () => {
            try {
                const buffer = Buffer.concat(data);
                // 保存图片
                // fs.writeFile("./image.jpg", buffer, (err) => {
                //     if (!err) {
                //         res.end("ok");
                //     };
                // });
                const image = tf.node.decodeImage(buffer);
                nsfwLoad.then((model) => {
                    model.classify(image).then((predictions) => {
                        let data = {};
                        for (i of predictions) {
                            data[i['className']] = Math.trunc(i['probability'] * 10000) / 100;
                        };
                        res.end(JSON.stringify({
                            'code': 200,
                            'msg': '',
                            'data': data
                        }));
                    });
                });
            } catch (e) {
                res.end(JSON.stringify({
                    'code': 500,
                    'msg': e.message,
                    'data': ''
                }));
            };
        });
    } catch (e) {
        res.end(JSON.stringify({
            'code': 500,
            'msg': e.message,
            'data': ''
        }));
    };
});

app.use((req, res) => {
    res.end('https://github.com/qiaoshouzi/nsfwjs-api');
});

app.listen(port, () => {
    console.log("server run at ", port);
});