利用百度ai實現文本和圖片審覈

圖片描述

以前作平臺內容發佈審覈都是本身構建一套違禁詞庫,在代碼中利用詞庫判斷用戶發佈的內容,如今能夠使用百度ai api完成這個功能。接下來就簡單說下怎麼作吧:web

首先打開百度ai 開發平臺 註冊一個帳號:
圖片描述redis

註冊帳號,進入控制檯
圖片描述json

建立本身的應用,獲取apikey 和祕鑰
圖片描述api

進入文檔頁 文本審覈:
圖片描述app

圖像審覈:
圖片描述curl

文檔很詳細,實現用戶發佈內容審覈 圖片審覈仍是很方便簡單的。我沒有使用官方的sdk,簡單的整合了一下做爲練手,如下是我簡單的代碼demo:
圖片描述post

class Sentive
{
    protected $accessTokenUrl = 'https://aip.baidubce.com/oauth/2.0/token';//獲取token url
    protected $textUrl = 'https://aip.baidubce.com/rest/2.0/antispam/v2/spam';//文本審覈url
    protected $imgUrl = 'https://aip.baidubce.com/api/v1/solution/direct/img_censor';//圖片審覈url
    protected $avatarUrl = 'https://aip.baidubce.com/rest/2.0/solution/v1/face_audit';//頭像審覈url

    protected $grant_type;

    protected $client_id;

    protected $client_secret;

    function __construct()
    {
        $this->grant_type = 'client_credentials';
        $this->client_id = 'xxx';//API Key
        $this->client_secret = 'xxx';//Secret Key
    }

    static function request($url = '', $param = '')
    {
        if (empty($url) || empty($param)) {
            return false;
        }

        $postUrl = $url;
        $curlPost = $param;
        $curl = curl_init();//初始化curl
        curl_setopt($curl, CURLOPT_URL, $postUrl);//抓取指定網頁
        curl_setopt($curl, CURLOPT_HEADER, 0);//設置header
        curl_setopt($curl, CURLOPT_RETURNTRANSFER, 1);//要求結果爲字符串且輸出到屏幕上
        curl_setopt($curl, CURLOPT_POST, 1);//post提交方式
        curl_setopt($curl, CURLOPT_POSTFIELDS, $curlPost);
        $data = curl_exec($curl);//運行curl
        curl_close($curl);


        return $data;
    }

    static function request_post($url = '', $param = array(), $type)
    {
        if (empty($url) || empty($param)) {
            return false;
        }

        $postUrl = $url;
        $curlPost = $param;
        $curl = curl_init();
        curl_setopt($curl, CURLOPT_URL, $postUrl);
        curl_setopt($curl, CURLOPT_HEADER, 0);
        // 要求結果爲字符串
        curl_setopt($curl, CURLOPT_RETURNTRANSFER, 1);
        // post方式
        curl_setopt($curl, CURLOPT_POST, 1);
        curl_setopt($curl, CURLOPT_SSL_VERIFYPEER, false);
        curl_setopt($curl, CURLOPT_POSTFIELDS, $curlPost);
        if ($type == "text") {
            curl_setopt($curl, CURLOPT_HTTPHEADER, array('Content-Type: application/x-www-form-urlencoded'));
        } else {
            curl_setopt($curl, CURLOPT_HTTPHEADER, array('Content-Type: application/json;charset=utf-8'));
        }
        curl_setopt($curl, CURLINFO_HEADER_OUT, true);
        $data = curl_exec($curl);
        $code = curl_getinfo($curl, CURLINFO_HTTP_CODE);

        if ($code === 0) {
            throw new \Exception(curl_error($curl));
        }
        curl_close($curl);
        return $data;
    }

    //獲取token
    public function getToken()
    {
        new Redis();
        $post_data['grant_type'] = $this->grant_type;
        $post_data['client_id'] = $this->client_id;
        $post_data['client_secret'] = $this->client_secret;
        $o = "";
        foreach ($post_data as $k => $v) {
            $o .= "$k=" . urlencode($v) . "&";
        }
        $post_data = substr($o, 0, -1);
        $res = self::request($this->accessTokenUrl, $post_data);
        $redis->setkey("filterToken", json_decode($res, true)['access_token']);
        return json_decode($res, true)['access_token'];
    }

    //文本審覈
    public function textVerify($data)
    {
        new Redis();
        $token = $redis->get("filterToken");
        if (empty($token)) {
            $token = $this->getToken();
        }
        $curl = $this->textUrl . "?access_token=" . $token;
        $result = self::request_post($curl, $data, "text");
        return json_decode($result, true);
    }

    //圖片審覈
    public function imgVerify($img)
    {
        $redis = new Redis();
        $token = $redis->get("filterToken");
        if (empty($token)) {
            $token = $this->getToken();
        }
        $curl = $this->imgUrl . "?access_token=" . $token;
        $bodys = array(
            'image' => $img,
            'scenes' => array("ocr",
                "face", "public", "politician", "antiporn", "terror", "webimage", "disgust",
                'watermark')
        );
        $bodys = json_encode($bodys);
        $result = self::request_post($curl, $bodys, "img");
        return json_decode($result, true);
    }


    //頭像審覈
    public function avatarVerify($img)
    {
        $redis = new Redis();
        $token = $redis->get("filterToken");
        if (empty($token)) {
            $token = $this->getToken();
        }
        $curl = $this->avatarUrl . "?access_token=" . $token;
        $bodys = array(
            "configId" => "1",
            "images" => $img
        );
        $result = self::request_post($curl, $bodys, "text");
        return json_decode($result, true);
    }

}
相關文章
相關標籤/搜索