Unity 百度AI实现Logo商标识别

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
using System;
/// <summary>
/// Logo识别响应数据结构
/// </summary>
[Serializable]
public class LogoRecognition
{
    /// <summary>
    /// 请求标识码，随机数，唯一
    /// </summary>
    public float log_id;
    /// <summary>
    /// 返回结果数目，即：result数组中元素个数
    /// </summary>
    public int result_num;
    /// <summary>
    /// 返回结果数组，每一项为一个识别出的logo
    /// </summary>
    public LogoRecognitionResult[] result;
}
 
[Serializable]
public class LogoRecognitionResult
{
    /// <summary>
    /// 位置信息
    /// </summary>
    public LogoRecognitionResultLocation location;
    /// <summary>
    /// 识别的品牌名称
    /// </summary>
    public string name;
    /// <summary>
    /// 分类结果置信度（0--1.0）
    /// </summary>
    public float probability;
    /// <summary>
    /// type=0为1千种高优商标识别结果;type=1为2万类logo库的结果；其它type为自定义logo库结果    
    /// </summary>
    public int type;
}
 
/// <summary>
/// 位置信息
/// </summary>
[Serializable]
public class LogoRecognitionResultLocation
{
    /// <summary>
    /// 左起像素位置
    /// </summary>
    public float left;
    /// <summary>
    /// 上起像素位置
    /// </summary>
    public float top;
    /// <summary>
    /// 像素宽
    /// </summary>
    public float width;
    /// <summary>
    /// 像素高
    /// </summary>
    public float height;
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
using System;
using UnityEngine;
using Newtonsoft.Json;
using System.Collections.Generic;
 
/// <summary>
/// 图像识别
/// </summary>
public class ImageRecognition
{
    //以下信息于百度开发者中心控制台创建应用获取
    private const string appID = "";
    private const string apiKey = "";
    private const string secretKey = "";
 
    /// <summary>
    /// Logo商标识别
    /// </summary>
    /// <param name="bytes">Logo图片字节数据</param>
    /// <param name="customLib">是否只使用自定义logo库的结果，默认false：返回自定义库+默认库的识别结果</param>
    /// <returns></returns>
    public static LogoRecognition Logo(byte[] bytes, bool customLib = false)
    {
        var client = new Baidu.Aip.ImageClassify.ImageClassify(apiKey, secretKey);
        try
        {
            var options = new Dictionary<string, object>
            {
                { "custom_lib", customLib}
            };
            var response = client.LogoSearch(bytes, options);
            LogoRecognition logoRecognition = JsonConvert.DeserializeObject<LogoRecognition>(response.ToString());
            return logoRecognition;
        }
        catch (Exception error)
        {
            Debug.LogError(error);
        }
        return null;
    }
}

1
2
3
4
5
6
7
8
9
10
using System.IO;
using UnityEngine;
 
public class Example : MonoBehaviour
{
    private void Start()
    {
        ImageRecognition.Logo(File.ReadAllBytes(Application.dataPath + "/Picture.jpg"));
    }
}