Unity實現(xiàn)車型識別的示例代碼

更新時間：2022年01月04日 10:16:47 作者：CoderZ1010

這篇文章主要介紹了在Unity中接入百度AI，實現(xiàn)檢測一張車輛圖片的具體車型。即對于輸入的一張圖片（可正常解碼，且長寬比適宜），輸出圖片的車輛品牌及型號。需要的可以參考一下

接口介紹：

該請求用于檢測一張車輛圖片的具體車型。即對于輸入的一張圖片（可正常解碼，且長寬比適宜），輸出圖片的車輛品牌及型號。

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

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

查閱官方文檔，以下是車型識別接口返回數(shù)據(jù)參數(shù)詳情：

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

using System;
 
/// <summary>
/// 車型識別
/// </summary>
[Serializable]
public class CarDetect
{
    /// <summary>
    /// 唯一的log id 用于定位問題
    /// </summary>
    public int log_id;
    /// <summary>
    /// 顏色
    /// </summary>
    public string color_result;
    /// <summary>
    /// 車型識別結(jié)果數(shù)組
    /// </summary>
    public CarDetectResult[] result;
}
 
[Serializable]
public class CarDetectResult
{
    /// <summary>
    /// 車型名稱
    /// </summary>
    public string name;
    /// <summary>
    /// 置信度
    /// </summary>
    public float score;
    /// <summary>
    /// 年份
    /// </summary>
    public string year;
    /// <summary>
    /// 對應(yīng)識別結(jié)果的百科詞條名稱
    /// </summary>
    public BaikeInfo baike_info;
    /// <summary>
    /// 車在圖片中的位置信息
    /// </summary>
    public string location_result;
}
 
[Serializable]
public class BaikeInfo
{
    /// <summary>
    /// 對應(yīng)識別結(jié)果百度百科頁面鏈接
    /// </summary>
    public string baike_url;
    /// <summary>
    /// 對應(yīng)識別結(jié)果百科圖片鏈接
    /// </summary>
    public string image_url;
    /// <summary>
    /// 對應(yīng)識別結(jié)果百科內(nèi)容描述
    /// </summary>
    public string description;
}

下載C# SDK：

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

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

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

using System;
using UnityEngine;
using Newtonsoft.Json;
using System.Collections.Generic;
 
/// <summary>
/// 圖像識別
/// </summary>
public class ImageRecognition
{
    //以下信息于百度開發(fā)者中心控制臺創(chuàng)建應(yīng)用獲取
    private const string appID = "";
    private const string apiKey = "";
    private const string secretKey = "";
 
    /// <summary>
    /// 車型識別
    /// </summary>
    /// <param name="bytes">圖片字節(jié)數(shù)據(jù)</param>
    /// <param name="topNum">返回預(yù)測得分top結(jié)果數(shù)</param>
    /// <param name="baikeNum">返回百科信息的結(jié)果數(shù)</param>
    /// <returns></returns>
    public static CarDetect Car(byte[] bytes, int topNum = 5, int baikeNum = 0)
    {
        var client = new Baidu.Aip.ImageClassify.ImageClassify(apiKey, secretKey);
        try
        {
            var options = new Dictionary<string, object>
            {
                {"top_num", topNum },
                {"baike_num", baikeNum }
            };
            var response = client.CarDetect(bytes, options);
            CarDetect carDetect = JsonUtility.FromJson<CarDetect>(response.ToString());
            return carDetect;
        }
        catch(Exception error)
        {
            Debug.LogError(error);
        }
        return null;
    }
}

測試圖片：

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