基于百度AI做内容审核

超、凢脫俗 2022-05-10 12:08 220阅读 0赞

**先哭诉一下，百度提供的API和当前版本不对称，让在下头疼了一个多小时，在此作文为大家使用提供参考。**

## 一、引入sdk ##

<dependency>
                <groupId>com.baidu.aip</groupId>
                <artifactId>java-sdk</artifactId>
                <version>LATEST</version>
            </dependency>

## 二、初始化应用参数 ##

private static final String APP_ID = "xxxx";
    
        private static final String API_KEY = "xxxx";
    
        private static final String SECRET_KEY = "xxxx";
    
        /**图片内容审核客户端*/
        private static AipContentCensor contentCensorClient;

/**
         * 初始化Client
         */
        private static void getClient() {
    
            //初始化图片审核客户端
            contentCensorClient = new AipContentCensor(APP_ID, API_KEY, SECRET_KEY);
            contentCensorClient.setConnectionTimeoutInMillis(2000);
            contentCensorClient.setSocketTimeoutInMillis(60000);
    
            // 初始化一个AipOcr
            ocrClient = new AipOcr(APP_ID, API_KEY, SECRET_KEY);
            ocrClient.setConnectionTimeoutInMillis(2000);
            ocrClient.setSocketTimeoutInMillis(60000);
        }

三、审核文本内容

/**
         * 识别文本中的敏感词汇(需要对百度API增强)
         *
         * @param text
         */
        public static boolean discernSensitiveWords(String text, HashMap<String, String> options)   {
            String resp;
            try {
                //防止出现特殊符号，制造异常
                resp = contentCensorClient.antiSpam(URLDecoder.decode(text, "UTF-8"), options).toString();
            } catch (UnsupportedEncodingException e) {
                return false;
            }
            //标识审核是否通过的结果所在未知
            int len = resp.lastIndexOf("m\":") + 3;
            String spam = resp.substring(len,len + 1);
            //System.out.println(spam);
            if(spam.equals("0")) {
                return true;
            }else {
                return false;
            }
        }

四、审核图形内容

/**
         * 识别图中的敏感内容(需要对百度API增强)
         *
         * @param image
         */
        public static boolean discernSensitiveImage(byte[] image) throws IOException {
    
            // 参数为本地图片文件二进制数组
            String resp = contentCensorClient.imageCensorUserDefined(image, null).toString();
            //截取conclusionType的值
            String result = resp.substring(resp.length() - 2,resp.length()-1);
            if (result.equals("1")) {
                return true;
            }else  {
                return false;
            }
    
        }

五、测试

![70][]

public static void main(String[] args) throws IOException {
            //文本内容审核
            boolean result1 = discernSensitiveWords("共产党", null);
            System.out.println("文本内容是否通过：" + result1);
            System.out.println(">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>");
            //图片审核
            InputStream inputStream = FileUtils.getInputStream("/file/test.jpg");
            byte[] b = new byte[inputStream.available()];
            inputStream.read(b);
            boolean result2 = discernSensitiveImage(b);
            System.out.println("图片内容是否通过：" + result2);
        }

![70 1][]

[70]: /images/20220510/16429c5bd19c4179beb7edf44a2e1043.png
[70 1]: /images/20220510/775b5853014744cc96fc798186c1b9da.png