首頁 > 軟體

Unity實現菜品識別的範例程式碼

2022-02-11 16:00:15

介面介紹:

識別超過9千種菜品,支援客戶建立屬於自己的菜品相簿,可準確識別圖片中的菜品名稱、位置、卡路里資訊,並獲取百科資訊,適用於多種客戶識別菜品的業務場景中。

建立應用:     

在產品服務中搜尋影象識別,建立應用,獲取AppID、APIKey、SecretKey資訊:

查閱官方檔案,以下是菜品識別介面返回資料引數詳情:

定義資料結構:

using System;
 
/// <summary>
/// 菜品識別
/// </summary>
[Serializable]
public class DishRecognition 
{
    /// <summary>
    /// 唯一的log id,用於問題定位
    /// </summary>
    public float log_id;
    /// <summary>
    /// 返回結果數目,及result陣列中的元素個數
    /// </summary>
    public int result_num;
    /// <summary>
    /// 菜品識別結果陣列
    /// </summary>
    public DishRecognitionResult[] result;
}
 
/// <summary>
/// 菜品識別結果
/// </summary>
[Serializable]
public class DishRecognitionResult
{
    /// <summary>
    /// 菜名
    /// </summary>
    public string name;
    /// <summary>
    /// 卡路里,每100g的卡路里含量
    /// </summary>
    public float calorie;
    /// <summary>
    /// 識別結果中每一行的置信度值,0-1
    /// </summary>
    public float probability;
    /// <summary>
    /// 百科詞條
    /// </summary>
    public BaikeInfo baike_info;
}
 
[Serializable]
public class BaikeInfo
{
    /// <summary>
    /// 對應識別結果百度百科頁面連結
    /// </summary>
    public string baike_url;
    /// <summary>
    /// 對應識別結果百科圖片連結
    /// </summary>
    public string image_url;
    /// <summary>
    /// 對應識別結果百科內容描述
    /// </summary>
    public string description;
}

下載C# SDK:

下載完成後將AipSdk.dll動態庫匯入到Unity中:

以下是呼叫介面時傳入的引數詳情:

封裝呼叫函數: 

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結果數,預設為5</param>
    /// <param name="filterThreshold">預設0.95,可以通過該引數調節識別效果,降低非菜識別率.</param>
    /// <param name="baikeNum">    返回百科資訊的結果數,預設不返回</param>
    /// <returns></returns>
    public static DishRecognition Dish(byte[] bytes, int topNum = 5, float filterThreshold = 0.95f, int baikeNum = 0)
    {
        var client = new Baidu.Aip.ImageClassify.ImageClassify(apiKey, secretKey);
        try
        {
            var options = new Dictionary<string, object>
            {
                { "top_num", topNum },
                { "filter_threshold", Mathf.Clamp01(filterThreshold)},
                { "baike_num", baikeNum}
            };
            var response = client.DishDetect(bytes, options);
            DishRecognition dishRecognition = JsonConvert.DeserializeObject<DishRecognition>(response.ToString());
            return dishRecognition;
        }
        catch (Exception error)
        {
            Debug.LogError(error);
        }
        return null;
    }
}

測試圖片:

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

到此這篇關於Unity實現菜品識別的範例程式碼的文章就介紹到這了,更多相關Unity菜品識別內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com