Unity接入百度AI實現通用物體和場景識別

Posted on by

接口介紹：

可識別超過10萬類常見物體和場景，接口返回大類及細分類的名稱，並支持獲取識別結果對應的百科信息；還可使用EasyDL定制訓練平臺，定制識別分類標簽。廣泛適用於圖像或視頻內容分析、拍照識圖等業務場景。

創建應用：     

在產品服務中搜索圖像識別，創建應用，獲取AppID、APIKey、SecretKey信息：

查閱官方文檔，以下是通用物體和場景識別接口返回數據參數詳情：

定義數據結構：

using System;
 
/// <summary>
/// 通用物體和場景識別
/// </summary>
[Serializable]
public class AdvancedGeneralRecognition
{
    /// <summary>
    /// 唯一的log id，用於問題定位
    /// </summary>
    public float log_id;
    /// <summary>
    /// 返回結果數目，及result數組中的元素個數
    /// </summary>
    public int result_num;
    /// <summary>
    /// 返回結果數組
    /// </summary>
    public AdvancedGeneralRecognitionResult[] result;
}
 
/// <summary>
/// 通用物體和場景識別結果
/// </summary>
[Serializable]
public class AdvancedGeneralRecognitionResult
{
    /// <summary>
    /// 圖片中的物體或場景名稱
    /// </summary>
    public string keyword;
    /// <summary>
    /// 置信度，0-1
    /// </summary>
    public float score;
    /// <summary>
    /// 識別結果的上層標簽，有部分錢幣、動漫、煙酒等tag無上層標簽
    /// </summary>
    public string root;
    /// <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="baikeNum">返回百科信息的結果數，默認不返回</param>
    /// <returns></returns>
    public static AdvancedGeneralRecognition AdvancedGeneral(byte[] bytes, int baikeNum = 0)
    {
        var client = new Baidu.Aip.ImageClassify.ImageClassify(apiKey, secretKey);
        try
        {
            var options = new Dictionary<string, object>
            {
                { "baike_num", baikeNum },
            };
            var response = client.AdvancedGeneral(bytes, options);
            AdvancedGeneralRecognition recognition = JsonConvert.DeserializeObject<AdvancedGeneralRecognition>(response.ToString());
            return recognition;
        }
        catch (Exception error)
        {
            Debug.LogError(error);
        }
        return null;
    }
}

測試圖片：

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

到此這篇關於Unity接入百度AI實現通用物體和場景識別的文章就介紹到這瞭,更多相關Unity物體和場景識別內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：