阿里云图像搜索Quick Start

概述

图像搜索服务(Image Search)是以深度学习和大规模机器学习技术为核心,通过图像识别和搜索功能,实现以图搜图的智能图像搜索产品。图像搜索服务在基于图像识别技术基础上,结合不同行业应用和业务场景,帮助用户实现相同或相似图片搜索。因为导入到实例的图片实例本身并不存储原始图片,所以在使用查询功能的时候,并不能直接返回原始图片或者可以直接访问的图片URL,需要使用者根据查询反馈的picname等信息自行到自己的图库中反查原始图片,这样给使用带来了很大的不便。

针对这种情况,提供一种解决办法,利用添加图片时候支持的属性:CustomContent,直接设置该值为添加图片可以公网访问的URL,这样在查询的时候就直接可以通过Response Auctions对象的CustomContent属性获取图片的URL,当然前提是本身加入的图片提供公网访问URL。

操作步骤

1、到图像搜索控制台创建实例

_

2、获取测试需要的相关参数

_

3、pom.xml

  <dependencies>
        <dependency>
            <groupId>com.aliyun</groupId>
            <artifactId>aliyun-java-sdk-imagesearch</artifactId>
            <version>2.0.0</version>
        </dependency>
        <dependency>
            <groupId>com.aliyun</groupId>
            <artifactId>aliyun-java-sdk-core</artifactId>
            <version>[4.3.2,5.0.0)</version>
        </dependency>
    </dependencies>

4、Code Sample

import com.aliyuncs.DefaultAcsClient;
import com.aliyuncs.IAcsClient;
import com.aliyuncs.exceptions.ClientException;
import com.aliyuncs.imagesearch.model.v20190325.AddImageRequest;
import com.aliyuncs.imagesearch.model.v20190325.AddImageResponse;
import com.aliyuncs.imagesearch.model.v20190325.SearchImageRequest;
import com.aliyuncs.imagesearch.model.v20190325.SearchImageResponse;
import com.aliyuncs.profile.DefaultProfile;
import com.aliyuncs.profile.IClientProfile;
import org.apache.commons.codec.binary.Base64;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;

public class AddAndSearchDemo {

    public static void main(String[] args) throws IOException, InterruptedException {

        String accessKeyId = "LTAIOZZgYX******";
        String accessSecret = "v7CjUJCMk7j9aKduMAQLjyCm******";
        String regionId = "cn-shanghai";
        String instanceName = "demop******";
        String productId = "shirt";
        String picName1 = "shirt_1";
        String picName2 = "shirt_2";
        String picURL1 = "https://timgsa.baidu.com/timg?image&quality=80&size=b9999_10000&sec=1577090672111&di=ecb316794a7d0eeb8dc9dd76d4ca78d0&imgtype=0&src=http%3A%2F%2Fimg12.360buyimg.com%2Fn0%2Fjfs%2Ft208%2F202%2F990939258%2F101503%2Fe8b76d70%2F53a13d75N1456d2d6.jpg";
        String picURL2 = "https://timgsa.baidu.com/timg?image&quality=80&size=b9999_10000&sec=1577090672122&di=539afa0415b469da9bb57daf94eff8e7&imgtype=0&src=http%3A%2F%2Fimg12.360buyimg.com%2Fn1%2Fjfs%2Ft877%2F18%2F399371448%2F100857%2Fd906afbb%2F551d5285Nd29ae9aa.jpg";
        String picPath = "C:\\******\\main\\resources\\test.jpg"; // 待查询图片的路径

        DefaultProfile.addEndpoint( regionId, "ImageSearch", "imagesearch.cn-shanghai.aliyuncs.com");
        IClientProfile profile = DefaultProfile.getProfile(regionId, accessKeyId, accessSecret);

        IAcsClient client = new DefaultAcsClient(profile);

        // 添加图片到实例
        addPic(client,instanceName,productId,picName1,picURL1);
        Thread.sleep(1000);// 防止出现flow control
        addPic(client,instanceName,productId,picName2,picURL2);

        // 查询图片
        searchPic(client, instanceName, picPath);
        System.out.println("测试结束");
    }

    /**
     * 添加图片到实例
     * @param client 实例对象
     * @param instanceName 实例名称
     * @param productId 商品Id
     * @param picName 图片名称
     * @param picURL 图片URL
     * @throws IOException
     */
    public static void addPic(IAcsClient client, String instanceName, String productId, String picName, String picURL) throws IOException {
        // 增加图片
        AddImageRequest addImageRequest = new AddImageRequest();
        // 必填,图像搜索实例名称。
        addImageRequest.setInstanceName(instanceName);
        // 必填,商品id,最多支持 512个字符。
        // 一个商品可有多张图片。
        addImageRequest.setProductId(productId);
        // 必填,图片名称,最多支持 512个字符。
        // 1. ProductId + PicName唯一确定一张图片。
        // 2. 如果多次添加图片具有相同的ProductId + PicName,以最后一次添加为准,前面添加的的图片将被覆盖。
        addImageRequest.setPicName(picName);
        // request.setStrAttr("");// 测试
        // 选填,图片类目。
        // 1. 对于商品搜索:若设置类目,则以设置的为准;若不设置类目,将由系统进行类目预测,预测的类目结果可在Response中获取 。
        // 2. 对于布料、商标、通用搜索:不论是否设置类目,系统会将类目设置为88888888。
        addImageRequest.setCategoryId(1);
        Base64 base64 = new Base64();
        String encodePicContent = base64.encodeToString(getFileFromUrl(picURL));
        // 必填,图片内容,Base64编码。
        // 最多支持 2MB大小图片以及5s的传输等待时间。当前仅支持jpg和png格式图片;
        // 对于商品、商标、通用图片搜索,图片长和宽的像素必须都大于等于200,并且小于等于1024;
        // 对于布料搜索,图片长和宽的像素必须都大于等于448,并且小于等于1024;
        // 图像中不能带有旋转信息。
        addImageRequest.setPicContent(encodePicContent);
        // 选填,是否需要进行主体识别,默认为true。
        // 1.为true时,由系统进行主体识别,以识别的主体进行搜索,主体识别结果可在Response中获取。
        // 2. 为false时,则不进行主体识别,以整张图进行搜索。
        // 3.对于布料图片搜索,此参数会被忽略,系统会以整张图进行搜索。
        addImageRequest.setCrop(true);
        // 选填,图片的主体区域,格式为 x1,x2,y1,y2, 其中 x1,y1 是左上角的点,x2,y2是右下角的点。
        // 若用户设置了Region,则不论Crop参数为何值,都将以用户输入Region进行搜索。
        // 对于布料图片搜索,此参数会被忽略,系统会以整张图进行搜索。
//        request.setRegion("280,486,232,351");
        // 选填,整数类型属性,可用于查询时过滤,查询时会返回该字段。
        //  例如不同的站点的图片/不同用户的图片,可以设置不同的IntAttr,查询时通过过滤来达到隔离的目的
        addImageRequest.setIntAttr(0);
        // 选填,字符串类型属性,最多支持 128个字符。可用于查询时过滤,查询时会返回该字段。
        addImageRequest.setStrAttr("demo");
        // 选填,用户自定义的内容,最多支持 4096个字符。
        // 查询时会返回该字段。例如可添加图片的描述等文本。
        // 这里设置为图片的公网地址URL,方便从查询结果中直接获取数据
        addImageRequest.setCustomContent(picURL);
        try {
            AddImageResponse addImageResponse = client.getAcsResponse(addImageRequest);
            System.out.println("Add Picture Result: " + addImageResponse.getSuccess());
        } catch (ClientException e) {
            // 抛出异常,例如参数无效,或者实例不可用等情况
            e.printStackTrace();
        }
    }

    /**
     * 使用本地图片到图片库中查询
     * @param client 实例对象
     * @param instanceName 实例名称
     * @param picPath 本地图片的路径
     */
    public static void searchPic(IAcsClient client, String instanceName, String picPath)
    {
        SearchImageRequest searchImageRequest = new SearchImageRequest();
        // 必填,图像搜索实例名称。
        searchImageRequest.setInstanceName(instanceName);
        // 选填,搜索类型,取值范围:
        // 1. SearchByPic(默认):根据图片搜索相似图片。
        // 2. SearchByName,根据已添加的图片搜索相似图片。
        // request.setType("SearchByPic");
        byte[] bytes2_search = getBytesFromPath(picPath);
        Base64 base64 = new Base64();
        String encodePicContent_seacrch = base64.encodeToString(bytes2_search);
        // 图片内容,Base64编码。最多支持 2MB大小图片以及5s的传输等待时间。当前仅支持jpg和png格式图片;
        // 对于商品、商标、通用图片搜索,图片长和宽的像素必须都大于等于200,并且小于等于1024;
        // 对于布料搜索,图片长和宽的像素必须都大于等于448,并且小于等于1024;
        // 图像中不能带有旋转信息。
        // 1. Type=SearchByPic时,必填
        // 2. Type=SearchByName时,无需填写。
        searchImageRequest.setPicContent(encodePicContent_seacrch);
        // 选填,商品类目。
        // 1. 对于商品搜索:若设置类目,则以设置的为准;若不设置类目,将由系统进行类目预测,预测的类目结果可在Response中获取 。
        // 2. 对于布料、商标、通用搜索:不论是否设置类目,系统会将类目设置为88888888。
        searchImageRequest.setCategoryId(1);
        // 选填,是否需要进行主体识别,默认为true。
        // 1.为true时,由系统进行主体识别,以识别的主体进行搜索,主体识别结果可在Response中获取。
        // 2. 为false时,则不进行主体识别,以整张图进行搜索。
        // 3.对于布料图片搜索,此参数会被忽略,系统会以整张图进行搜索。
        searchImageRequest.setCrop(false);
        // 选填,图片的主体区域,格式为 x1,x2,y1,y2, 其中 x1,y1 是左上角的点,x2,y2是右下角的点。
        // 若用户设置了Region,则不论Crop参数为何值,都将以用户输入Region进行搜索。
        // 3.对于布料图片搜索,此参数会被忽略,系统会以整张图进行搜索。
        // searchImageRequest.setRegion("280,486,232,351");
        // 选填,返回结果的数目。取值范围:1-100。默认值:10。
        searchImageRequest.setNum(2);
        // 选填,返回结果的起始位置。取值范围:0-499。默认值:0。
        // searchImageRequest.setStart(1);
        // 选填,过滤条件
        // int_attr支持的操作符有>、>=、<、<=、=,str_attr支持的操作符有=和!=,多个条件之支持AND和OR进行连接。
        // 示例:
        //  1. 根据IntAttr过滤结果,int_attr>=100
        //  2. 根据StrAttr过滤结果,str_attr!="value1"
        //  3. 根据IntAttr和StrAttr联合过滤结果,int_attr=1000 AND str_attr="value1"
        //   searchImageRequest.setFilter("int_attr=1");
        try {
            SearchImageResponse searchImageResponse = client.getAcsResponse(searchImageRequest);
            System.out.println("getCustomContent: " + searchImageResponse.getAuctions().get(0).getCustomContent());
            System.out.println("getCustomContent: " + searchImageResponse.getAuctions().get(1).getCustomContent());
        } catch (ClientException e) {
            e.printStackTrace();
        }
    }

    /**
     * 通过oss 中的URL的读取图片到Byte[]
     * @param url 公网可以访问的图片URL
     * @return byte[]
     * @throws IOException
     */
    private static byte[] getFileFromUrl(String url) throws IOException{
        URL urlConet = new URL(url);
        HttpURLConnection con = (HttpURLConnection)urlConet.openConnection();
        con.setRequestMethod("GET");
        con.setConnectTimeout(4 * 1000);
        InputStream inStream = con .getInputStream();//通过输入流获取图片数据
        ByteArrayOutputStream outStream = new ByteArrayOutputStream();
        byte[] buffer = new byte[2048];
        int len = 0;
        while( (len=inStream.read(buffer)) != -1 ){
            outStream.write(buffer, 0, len);
        }
        inStream.close();
        byte[] data =  outStream.toByteArray();
        return data;
    }

    /**
     * 从本地读取图片到byte[]
     * @param filePath 本地图片的路径
     * @return byte[]
     */
    private static byte[] getBytesFromPath(String filePath) {
        byte[] buffer = null;
        try {
            File file = new File(filePath);
            FileInputStream fis = new FileInputStream(file);
            // picture max size is 2MB
            ByteArrayOutputStream bos = new ByteArrayOutputStream(2000 * 1024);
            byte[] b = new byte[1000];
            int n;
            while ((n = fis.read(b)) != -1) {
                bos.write(b, 0, n);
            }
            fis.close();
            bos.close();
            buffer = bos.toByteArray();
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return buffer;
    }
}

备注:待查询图片下载地址

5、测试结果

Add Picture Result: true
Add Picture Result: true
getCustomContent: https://timgsa.baidu.com/timg?image&quality=80&size=b9999_10000&sec=1577090672111&di=ecb316794a7d0eeb8dc9dd76d4ca78d0&imgtype=0&src=http%3A%2F%2Fimg12.360buyimg.com%2Fn0%2Fjfs%2Ft208%2F202%2F990939258%2F101503%2Fe8b76d70%2F53a13d75N1456d2d6.jpg
getCustomContent: https://timgsa.baidu.com/timg?image&quality=80&size=b9999_10000&sec=1577090672122&di=539afa0415b469da9bb57daf94eff8e7&imgtype=0&src=http%3A%2F%2Fimg12.360buyimg.com%2Fn1%2Fjfs%2Ft877%2F18%2F399371448%2F100857%2Fd906afbb%2F551d5285Nd29ae9aa.jpg
测试结束

更多参考

Java SDK
新增图片 API
查询图片 API

猜你喜欢

转载自yq.aliyun.com/articles/740241