package com.javaweb.third.baidu.api; import com.javaweb.third.baidu.util.Base64Util; import com.javaweb.third.baidu.util.FileUtil; import com.javaweb.third.baidu.util.HttpUtil; import java.net.URLEncoder; /** * 图像识别 * @author wujiyue */ public class ImageClassifyAPI extends BaiduAi{ private static final String URL_advanced_general="https://aip.baidubce.com/rest/2.0/image-classify/v2/advanced_general"; private static final String URL_object_detect="https://aip.baidubce.com/rest/2.0/image-classify/v1/object_detect"; private static final String URL_dish="https://aip.baidubce.com/rest/2.0/image-classify/v2/dish"; private static final String URL_car="https://aip.baidubce.com/rest/2.0/image-classify/v1/car"; private static final String URL_logo="https://aip.baidubce.com/rest/2.0/image-classify/v2/logo"; private static final String URL_plant="https://aip.baidubce.com/rest/2.0/image-classify/v1/plant"; private static final String URL_animal="https://aip.baidubce.com/rest/2.0/image-classify/v1/animal"; //该请求用于通用物体及场景识别，即对于输入的一张图片（可正常解码，且长宽比适宜），输出图片中的多个物体及场景标签。 /** * { "result": [ { "score": 0.996184, "root": "商品-电脑办公", "keyword": "台式机" }, { "score": 0.799921, "root": "Logo", "keyword": "联想" }, { "score": 0.424538, "root": "商品-电脑办公", "keyword": "电脑" }, { "score": 0.213236, "root": "商品-电脑办公", "keyword": "服务器/工作站" }, { "score": 0.10919, "root": "商品-数码产品", "keyword": "台式电脑" } ], "result_num": 5, "log_id": 2689001477 } * @param path * @return */ public static String advanced_general(String path) { try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); String result = HttpUtil.post(URL_advanced_general, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //用户向服务请求检测图像中的主体位置。 public static String object_detect(String path,boolean withFace){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8"); if(withFace){ params+="&with_face=1"; }else{ params+="&with_face=0"; } String result = HttpUtil.post(URL_object_detect, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //该请求用于菜品识别。即对于输入的一张图片（可正常解码，且长宽比适宜），输出图片的菜品名称、卡路里信息、置信度。 /*{ "log_id": 2471272194, "result_num": 5, "result": [ { "name": "烧烤（串类）", "calorie": 333.33, "probability": 0.35874313116074 }, { "name": "鱿鱼", "calorie": 333.33, "probability": 0.20610593259335 }, { "name": "板筋", "calorie": 333.33, "probability": 0.15860831737518 }, { "name": "鸡脆骨", "calorie": 333.33, "probability": 0.077698558568954 }, { "name": "麻辣烫", "calorie": 333.33, "probability": 0.041968926787376 } ] } 识别失败： {"log_id": 2677811678621716503, "result_num": 1, "result": [{"calorie": "-1", "has_calorie": true, "name": "非菜", "probability": "0.202248"}]} */ public static String dish(String path,Integer topNum){ try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String params = URLEncoder.encode("image", "UTF-8") + "=" + URLEncoder.encode(imgStr, "UTF-8")+"&filter_threshold=0.95"; if(topNum!=null&&topNum>0){ params+="&top_num="+topNum; }else{ params+="&top_num="+5; } String result = HttpUtil.post(URL_dish, getAccessToken(), params); return result; } catch (Exception e) { e.printStackTrace(); } return null; } //车型识别 /** * * {"log_id": 6322863901722339584, "location_result": {"width": 729, "top": 193, "height": 302, "left": 132}, "result": [{"score": 0.99944090843201, "name": "宝骏560", "year": "2015-2017"}, {"score": 0.00022018732852302, "name": "宝骏730", "year": "2016-2017"}, {"score": 0.00013495099847205, "name": "东风风行风行SX6", "year": "2016"}, {"score": 1.4159925740387e-05, "name": "西雅特Ateca", "year": "无年份信息"}, {"score": 1.1880139027198e-05, "name": "东风风度MX5", "year": "2016-2017"}, {"score": 1.1864442967635e-05, "name": "大众途昂", "year": "2017"}, {"score": 9.8557611636352e-06, "name": "本田Pilot", "year": "无年份信息"}, {"score": 9.3120852397988e-06, "name": "开瑞K60", "year": "2017"}, {"score": 9.2856907940586e-06, "name": "标致3008", "year": "2013-2016"}, {"score": 8.5839219536865e-06, "name": "双龙途凌", "year": "2016-2017"}, {"score": 6.7547152866609e-06, "name": "江淮瑞风S系列", "year": "2017"}, {"score": 6.1060736697982e-06, "name": "北汽幻速幻速S6", "year": "2016-2017"}], "color_result": "白色"} * @param path * @param topNum * @return */ public static String car(String path,Integer topNum) { try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String imgParam = URLEncoder.encode(imgStr, "UTF-8"); String param = "image=" + imgParam; if(topNum!=null&&topNum>0){ param+="&top_num="+topNum; }else{ param+="&top_num="+5; } String result = HttpUtil.post(URL_car, getAccessToken(), param); return result; } catch (Exception e) { e.printStackTrace(); } return null; } /** * logo检测 * @param path * @param flag 是否只检索用户子库，true则只检索用户子库，false(默认)为检索底库+用户子库 * @return *{"log_id": 3510281299570976969, "result_num": 1, "result": [{"type": 0, "name": "香奈儿", "probability": 1, "location": {"width": 288, "top": 35, "left": 164, "height": 224}}]} */ public static String logo(String path,Boolean flag) { try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String imgParam = URLEncoder.encode(imgStr, "UTF-8"); String param = "image=" + imgParam; if(flag!=null && flag){ param += "&custom_lib=true"; }else{ param += "&custom_lib=false"; } String result = HttpUtil.post(URL_logo, getAccessToken(), param); return result; } catch (Exception e) { e.printStackTrace(); } return null; } /** * * @param path * @return * {"log_id": 8348105070592846270, "result": [{"score": 0.83211559057236, "name": "黑心金光菊"}, {"score": 0.10564804077148, "name": "金光菊"}, {"score": 0.025003267452121, "name": "连翘"}, {"score": 0.0059820651076734, "name": "虎眼金光菊"}, {"score": 0.0052708736620843, "name": "狗舌草"}]} */ public static String plant(String path) { try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String imgParam = URLEncoder.encode(imgStr, "UTF-8"); String param = "image=" + imgParam; String result = HttpUtil.post(URL_plant, getAccessToken(), param); return result; } catch (Exception e) { e.printStackTrace(); } return null; } /** * 动物识别 * @param path * @param top_num 返回预测得分top结果数，默认为6 * @return * {"log_id": 8619617754141201963, "result": [{"score": "0.97435", "name": "红鹳"}, {"score": "0.011475", "name": "红鹤"}]} */ public static String animal(String path,Integer top_num) { try { byte[] imgData = FileUtil.readFileByBytes(path); String imgStr = Base64Util.encode(imgData); String imgParam = URLEncoder.encode(imgStr, "UTF-8"); String param = "image=" + imgParam; if(top_num!=null&&top_num > 0){ param +="&top_num="+top_num; }else{ param +="&top_num=6"; } String result = HttpUtil.post(URL_animal, getAccessToken(), param); return result; } catch (Exception e) { e.printStackTrace(); } return null; } public static void main(String[] args) { /*String path="C:\\Users\\Administrator\\Desktop\\ge2.jpg"; String s=advanced_general(path); System.out.println(s);*/ /*String path="C:\\Users\\Administrator\\Desktop\\lc.jpg"; String s=dish(path,3); System.out.println(s);*/ /*String path="C:\\Users\\wjy\\Desktop\\dish.jpg"; String s=car(path,2); System.out.println(s);*/ /* String path="C:\\Users\\wjy\\Desktop\\car.jpg"; String s= logo(path, false); System.out.println(s);*/ /*String path="C:\\Users\\wjy\\Desktop\\plant.jpg"; String s= plant(path); System.out.println(s);*/ String path = "C:\\Users\\wjy\\Desktop\\car.jpg"; String s = object_detect(path,false); System.out.println(s); } }