当前位置:主页 > 软件编程 > .NET代码 >

Unity接入百度AI实现果蔬识别

时间:2022-12-13 09:30:59 | 栏目:.NET代码 | 点击:

接口介绍:

识别近千种水果和蔬菜的名称,适用于识别只含有一种果蔬的图片,可自定义返回识别结果数,适用于果蔬介绍相关的美食类APP中。

创建应用:     

在产品服务中搜索图像识别,创建应用,获取AppID、APIKey、SecretKey信息:

查阅官方文档,以下是果蔬识别接口返回数据参数详情:

定义数据结构:

using System;
 
/// <summary>
/// 果蔬识别
/// </summary>
[Serializable]
public class IngredientRecognition
{
    /// <summary>
    /// 唯一的log id,用于问题定位
    /// </summary>
    public float log_id;
    /// <summary>
    /// 返回结果数目,及result数组中的元素个数
    /// </summary>
    public int result_num;
    /// <summary>
    /// 识别结果数组
    /// </summary>
    public IngredientRecognitionResult[] result;
}
 
/// <summary>
/// 果蔬识别结果
/// </summary>
[Serializable]
public class IngredientRecognitionResult
{
    /// <summary>
    /// 食材名称
    /// </summary>
    public string name;
    /// <summary>
    /// 置信度
    /// </summary>
    public float score;
}

下载C# SDK:

 下载完成后将AipSdk.dll动态库导入到Unity中:

以下是调用接口时传入的参数详情:

在下载的SDK中并未发现通过url调用的重载函数,大概是官方文档未更新:

封装调用函数: 

using System;
using System.Collections.Generic;
using UnityEngine;
 
/// <summary>
/// 图像识别
/// </summary>
public class ImageRecognition 
{
    //以下信息于百度开发者中心控制台创建应用获取
    private const string appID = "";
    private const string apiKey = "";
    private const string secretKey = "";
 
    /// <summary>
    /// 果蔬识别
    /// </summary>
    /// <param name="bytes">图片字节数据</param>
    /// <param name="topNum">返回预测得分top结果数,如果为空或小于等于0默认为5;如果大于20默认20</param>
    /// <returns></returns>
    public static IngredientRecognition Ingredient(byte[] bytes, int topNum = 5)
    {
        var client = new Baidu.Aip.ImageClassify.ImageClassify(apiKey, secretKey);
        try
        {
            var options = new Dictionary<string, object>
            {
                { "top_num", topNum },
            };
            var response = client.Ingredient(bytes, options);
            IngredientRecognition ingredientRecognition = JsonConvert.DeserializeObject<IngredientRecognition>(response.ToString());
            return ingredientRecognition;
        }
        catch (Exception error)
        {
            Debug.LogError(error);
        }
        return null;
    }
}

测试图片:

using System.IO;
using UnityEngine;
 
public class Example : MonoBehaviour
{
    private void Start()
    {
        ImageRecognition.Ingredient(File.ReadAllBytes(Application.dataPath + "/Picture.jpg"));
    }
}

您可能感兴趣的文章:

相关文章