快捷導(dǎo)航

Unity實(shí)現(xiàn)植物識(shí)別示例詳解

更新時(shí)間：2022年01月05日 11:06:46 作者：CoderZ1010

這篇文章主要介紹了如何通過(guò)Unity接入百度AI實(shí)現(xiàn)植物識(shí)別，接口返回植物的名稱(chēng)，并支持獲取識(shí)別結(jié)果對(duì)應(yīng)的百科信息。感興趣的可以了解一下

接口介紹：

可識(shí)別超過(guò)2萬(wàn)種常見(jiàn)植物和近8千種花卉，接口返回植物的名稱(chēng)，并支持獲取識(shí)別結(jié)果對(duì)應(yīng)的百科信息；還可使用EasyDL定制訓(xùn)練平臺(tái)，定制識(shí)別植物種類(lèi)。適用于拍照識(shí)圖、幼教科普、圖像內(nèi)容分析等場(chǎng)景。

創(chuàng)建應(yīng)用：

在產(chǎn)品服務(wù)中搜索圖像識(shí)別，創(chuàng)建應(yīng)用，獲取AppID、APIKey、SecretKey信息：

查閱官方文檔，以下是植物識(shí)別接口返回?cái)?shù)據(jù)參數(shù)詳情：

定義數(shù)據(jù)結(jié)構(gòu)：

using System;
 
/// <summary>
/// 植物識(shí)別
/// </summary>
[Serializable]
public class PlantRecognition
{
    /// <summary>
    /// 唯一的log id，用于問(wèn)題定位
    /// </summary>
    public float log_id;
    /// <summary>
    /// 識(shí)別結(jié)果數(shù)組
    /// </summary>
    public PlantRecognition[] result;
}
 
[Serializable]
public class PlantRecognitionResult
{
    /// <summary>
    /// 植物名稱(chēng)
    /// </summary>
    public string name;
    /// <summary>
    /// 置信度
    /// </summary>
    public float score;
    /// <summary>
    /// 百科詞條
    /// </summary>
    public BaikeInfo baike_info;
}
 
[Serializable]
public class BaikeInfo
{
    /// <summary>
    /// 對(duì)應(yīng)識(shí)別結(jié)果百度百科頁(yè)面鏈接
    /// </summary>
    public string baike_url;
    /// <summary>
    /// 對(duì)應(yīng)識(shí)別結(jié)果百科圖片鏈接
    /// </summary>
    public string image_url;
    /// <summary>
    /// 對(duì)應(yīng)識(shí)別結(jié)果百科內(nèi)容描述
    /// </summary>
    public string description;
}

下載C# SDK：

下載完成后將AipSdk.dll動(dòng)態(tài)庫(kù)導(dǎo)入到Unity中：

以下是調(diào)用接口時(shí)傳入的參數(shù)詳情：

封裝調(diào)用函數(shù)：

using System;
using System.Collections.Generic;
using UnityEngine;
 
/// <summary>
/// 圖像識(shí)別
/// </summary>
public class ImageRecognition 
{
    //以下信息于百度開(kāi)發(fā)者中心控制臺(tái)創(chuàng)建應(yīng)用獲取
    private const string appID = "";
    private const string apiKey = "";
    private const string secretKey = "";
 
    /// <summary>
    /// 植物識(shí)別
    /// </summary>
    /// <param name="bytes">圖片字節(jié)數(shù)據(jù)</param>
    /// <param name="baikeNum">返回百科信息的結(jié)果數(shù)，默認(rèn)不返回</param>
    /// <returns></returns>
    public static PlantRecognition Plant(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.PlantDetect(bytes, options);
            Debug.Log(response);
            PlantRecognition plantRecognition = JsonConvert.DeserializeObject<PlantRecognition>(response.ToString());
            return plantRecognition;
        }
        catch (Exception error)
        {
            Debug.LogError(error);
        }
        return null;
    }
}

測(cè)試圖片：

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