【AI】图像识别-物体检测-百度AI-EasyDL-NodeJS

var https = require('https')
var express = require('express');
var app = express();
var bodyParser = require("body-parser");
var request = require('request')
var fs = require('fs'); // 引入fs模块

//访问端口
var port = 9999

//接受post数据
app.use(bodyParser.json({
    "limit": "100mb"
}));
app.use(bodyParser.urlencoded({
    limit: '50mb',
    extended: true
}));

//设置跨域访问
app.all('*', function (req, res, next) {
    res.header("Access-Control-Allow-Origin", "*");
    res.header("Access-Control-Allow-Headers", "X-Requested-With");
    res.header("Access-Control-Allow-Methods", "PUT,POST,GET,DELETE,OPTIONS");
    res.header("X-Powered-By", ' 3.2.1')
    res.header("Content-Type", "application/json;charset=utf-8");
    next();
});
var qs = require('querystring');

const param = qs.stringify({
    'grant_type': 'client_credentials',
    'client_id': 'XXXXX',
    'client_secret': 'XXXXXX'
});

function log(data)
{
    fs.writeFile('imgdata.txt', data, {
        'flag': 'a'
    }, function (err) {
        if (err) {
            throw err;
        }

        console.log('Hello.');

    });
}

app.post('/XXXXDetectV1', function (req, res) {

    var imgdata = req.body.imgdata; //body query
    // console.log(imgdata)
    imgdata = imgdata.replace("data:image/png;base64,","")
    log(imgdata)
    https.get({
            hostname: 'aip.baidubce.com',
            path: '/oauth/2.0/token?' + param,
            agent: false
        },
        function (res2) {
            // 在标准输出中查看运行结果
            // res2.pipe(process.stdout);
            // console.log('statusCode:', res2.statusCode);
            // console.log('headers:', res2.headers);

            res2.on('data', (d) => {
                console.log("认证成功")
                // console.log('' + d); //将buffer转为字符串或者使用d.toString()
                let b = JSON.parse('' + d); //将buffer转成JSON
                // console.log(b)
                var access_token = b["access_token"]
                console.log(access_token)
                // res.end(JSON.stringify(b));
                data = {
                    'image': imgdata
                }

                request_url = "https://aip.baidubce.com/rpc/2.0/ai_custom/v1/detection/XXXXDetectV1" + "?access_token=" + access_token

                request({
                    url: request_url,
                    method: "POST",
                    json: true,
                    headers: {
                        "content-type": "application/json",
                    },
                    body: data
                }, function (error, response, body) {
                    console.log(error)
                    // console.log(response)
                    // console.log(body)
                    if (!error && response.statusCode == 200) {
                        console.log(body) // 请求成功的处理逻辑
                        res.end(JSON.stringify(body));
                    }
                });

            });

        }
    );

})

//启动服务---------------------------------------------------------------------------------------------
var server = app.listen(port, function () {

    var host = server.address().address
    var port = server.address().port

    console.log("应用实例，访问地址为 http://%s:%s", host, port)

})