11 KiB
將影像分類 - Wio Terminal
在這部分課程中,你將把相機捕捉的影像發送到 Custom Vision 服務進行分類。
將影像分類
Custom Vision 服務提供了一個 REST API,Wio Terminal 可以使用它來分類影像。這個 REST API 是通過 HTTPS 連線(安全的 HTTP 連線)訪問的。
在與 HTTPS 端點互動時,客戶端程式碼需要向被訪問的伺服器請求公鑰憑證,並使用該憑證加密傳輸的資料。你的網頁瀏覽器會自動完成這個過程,但微控制器不會。因此,你需要手動請求這個憑證,並用它來建立與 REST API 的安全連線。這些憑證通常不會更改,因此一旦獲取憑證,就可以將其硬編碼到應用程式中。
這些憑證包含公鑰,無需保密。你可以在原始碼中使用它們,並公開分享,例如在 GitHub 上。
任務 - 設置 SSL 客戶端
-
如果尚未打開,請打開
fruit-quality-detector
應用程式專案。 -
打開
config.h
標頭檔案,並新增以下內容:const char *CERTIFICATE = "-----BEGIN CERTIFICATE-----\r\n" "MIIF8zCCBNugAwIBAgIQAueRcfuAIek/4tmDg0xQwDANBgkqhkiG9w0BAQwFADBh\r\n" "MQswCQYDVQQGEwJVUzEVMBMGA1UEChMMRGlnaUNlcnQgSW5jMRkwFwYDVQQLExB3\r\n" "d3cuZGlnaWNlcnQuY29tMSAwHgYDVQQDExdEaWdpQ2VydCBHbG9iYWwgUm9vdCBH\r\n" "MjAeFw0yMDA3MjkxMjMwMDBaFw0yNDA2MjcyMzU5NTlaMFkxCzAJBgNVBAYTAlVT\r\n" "MR4wHAYDVQQKExVNaWNyb3NvZnQgQ29ycG9yYXRpb24xKjAoBgNVBAMTIU1pY3Jv\r\n" "c29mdCBBenVyZSBUTFMgSXNzdWluZyBDQSAwNjCCAiIwDQYJKoZIhvcNAQEBBQAD\r\n" "ggIPADCCAgoCggIBALVGARl56bx3KBUSGuPc4H5uoNFkFH4e7pvTCxRi4j/+z+Xb\r\n" "wjEz+5CipDOqjx9/jWjskL5dk7PaQkzItidsAAnDCW1leZBOIi68Lff1bjTeZgMY\r\n" "iwdRd3Y39b/lcGpiuP2d23W95YHkMMT8IlWosYIX0f4kYb62rphyfnAjYb/4Od99\r\n" "ThnhlAxGtfvSbXcBVIKCYfZgqRvV+5lReUnd1aNjRYVzPOoifgSx2fRyy1+pO1Uz\r\n" "aMMNnIOE71bVYW0A1hr19w7kOb0KkJXoALTDDj1ukUEDqQuBfBxReL5mXiu1O7WG\r\n" "0vltg0VZ/SZzctBsdBlx1BkmWYBW261KZgBivrql5ELTKKd8qgtHcLQA5fl6JB0Q\r\n" "gs5XDaWehN86Gps5JW8ArjGtjcWAIP+X8CQaWfaCnuRm6Bk/03PQWhgdi84qwA0s\r\n" "sRfFJwHUPTNSnE8EiGVk2frt0u8PG1pwSQsFuNJfcYIHEv1vOzP7uEOuDydsmCjh\r\n" "lxuoK2n5/2aVR3BMTu+p4+gl8alXoBycyLmj3J/PUgqD8SL5fTCUegGsdia/Sa60\r\n" "N2oV7vQ17wjMN+LXa2rjj/b4ZlZgXVojDmAjDwIRdDUujQu0RVsJqFLMzSIHpp2C\r\n" "Zp7mIoLrySay2YYBu7SiNwL95X6He2kS8eefBBHjzwW/9FxGqry57i71c2cDAgMB\r\n" "AAGjggGtMIIBqTAdBgNVHQ4EFgQU1cFnOsKjnfR3UltZEjgp5lVou6UwHwYDVR0j\r\n" "BBgwFoAUTiJUIBiV5uNu5g/6+rkS7QYXjzkwDgYDVR0PAQH/BAQDAgGGMB0GA1Ud\r\n" "JQQWMBQGCCsGAQUFBwMBBggrBgEFBQcDAjASBgNVHRMBAf8ECDAGAQH/AgEAMHYG\r\n" "CCsGAQUFBwEBBGowaDAkBggrBgEFBQcwAYYYaHR0cDovL29jc3AuZGlnaWNlcnQu\r\n" "Y29tMEAGCCsGAQUFBzAChjRodHRwOi8vY2FjZXJ0cy5kaWdpY2VydC5jb20vRGln\r\n" "aUNlcnRHbG9iYWxSb290RzIuY3J0MHsGA1UdHwR0MHIwN6A1oDOGMWh0dHA6Ly9j\r\n" "cmwzLmRpZ2ljZXJ0LmNvbS9EaWdpQ2VydEdsb2JhbFJvb3RHMi5jcmwwN6A1oDOG\r\n" "MWh0dHA6Ly9jcmw0LmRpZ2ljZXJ0LmNvbS9EaWdpQ2VydEdsb2JhbFJvb3RHMi5j\r\n" "cmwwHQYDVR0gBBYwFDAIBgZngQwBAgEwCAYGZ4EMAQICMBAGCSsGAQQBgjcVAQQD\r\n" "AgEAMA0GCSqGSIb3DQEBDAUAA4IBAQB2oWc93fB8esci/8esixj++N22meiGDjgF\r\n" "+rA2LUK5IOQOgcUSTGKSqF9lYfAxPjrqPjDCUPHCURv+26ad5P/BYtXtbmtxJWu+\r\n" "cS5BhMDPPeG3oPZwXRHBJFAkY4O4AF7RIAAUW6EzDflUoDHKv83zOiPfYGcpHc9s\r\n" "kxAInCedk7QSgXvMARjjOqdakor21DTmNIUotxo8kHv5hwRlGhBJwps6fEVi1Bt0\r\n" "trpM/3wYxlr473WSPUFZPgP1j519kLpWOJ8z09wxay+Br29irPcBYv0GMXlHqThy\r\n" "8y4m/HyTQeI2IMvMrQnwqPpY+rLIXyviI2vLoI+4xKE4Rn38ZZ8m\r\n" "-----END CERTIFICATE-----\r\n";
這是 Microsoft Azure DigiCert Global Root G2 憑證,它是許多 Azure 服務全球使用的憑證之一。
💁 要確認這是需要使用的憑證,可以在 macOS 或 Linux 上執行以下命令。如果你使用的是 Windows,可以通過 Windows Subsystem for Linux (WSL) 執行此命令:
openssl s_client -showcerts -verify 5 -connect api.cognitive.microsoft.com:443
輸出結果將列出 DigiCert Global Root G2 憑證。
-
打開
main.cpp
,新增以下 include 指令:#include <WiFiClientSecure.h>
-
在 include 指令下方,宣告一個
WifiClientSecure
的實例:WiFiClientSecure client;
這個類別包含用於通過 HTTPS 與網路端點通信的程式碼。
-
在
connectWiFi
方法中,將 WiFiClientSecure 設置為使用 DigiCert Global Root G2 憑證:client.setCACert(CERTIFICATE);
任務 - 將影像分類
-
在
platformio.ini
檔案的lib_deps
列表中新增以下內容作為額外一行:bblanchon/ArduinoJson @ 6.17.3
這會匯入 ArduinoJson,一個 Arduino 的 JSON 函式庫,將用於解碼來自 REST API 的 JSON 回應。
-
在
config.h
中,新增 Custom Vision 服務的預測 URL 和金鑰的常數:const char *PREDICTION_URL = "<PREDICTION_URL>"; const char *PREDICTION_KEY = "<PREDICTION_KEY>";
將
<PREDICTION_URL>
替換為 Custom Vision 的預測 URL,將<PREDICTION_KEY>
替換為預測金鑰。 -
在
main.cpp
中,新增 ArduinoJson 函式庫的 include 指令:#include <ArduinoJSON.h>
-
在
main.cpp
中,將以下函數新增到buttonPressed
函數上方:void classifyImage(byte *buffer, uint32_t length) { HTTPClient httpClient; httpClient.begin(client, PREDICTION_URL); httpClient.addHeader("Content-Type", "application/octet-stream"); httpClient.addHeader("Prediction-Key", PREDICTION_KEY); int httpResponseCode = httpClient.POST(buffer, length); if (httpResponseCode == 200) { String result = httpClient.getString(); DynamicJsonDocument doc(1024); deserializeJson(doc, result.c_str()); JsonObject obj = doc.as<JsonObject>(); JsonArray predictions = obj["predictions"].as<JsonArray>(); for(JsonVariant prediction : predictions) { String tag = prediction["tagName"].as<String>(); float probability = prediction["probability"].as<float>(); char buff[32]; sprintf(buff, "%s:\t%.2f%%", tag.c_str(), probability * 100.0); Serial.println(buff); } } httpClient.end(); }
這段程式碼首先宣告了一個
HTTPClient
,這是一個包含與 REST API 互動方法的類別。接著,它使用之前設置的 Azure 公鑰,通過WiFiClientSecure
實例連接到預測 URL。連接後,它會發送標頭——這些是即將對 REST API 發出的請求的相關資訊。
Content-Type
標頭表示 API 呼叫將發送原始二進位資料,Prediction-Key
標頭則傳遞 Custom Vision 的預測金鑰。接著,對 HTTP 客戶端發出 POST 請求,上傳一個位元組陣列。這個陣列包含了從相機捕捉的 JPEG 影像。
💁 POST 請求用於發送資料並獲取回應。還有其他請求類型,例如 GET 請求,用於檢索資料。GET 請求通常由你的網頁瀏覽器用來加載網頁。
POST 請求會返回一個回應狀態碼。這些狀態碼是定義良好的值,其中 200 表示 OK——POST 請求成功。
💁 你可以在 維基百科的 HTTP 狀態碼列表頁面 中查看所有回應狀態碼。
如果返回 200,則從 HTTP 客戶端讀取結果。這是一個來自 REST API 的文字回應,包含預測結果的 JSON 文件。JSON 的格式如下:
{ "id":"45d614d3-7d6f-47e9-8fa2-04f237366a16", "project":"135607e5-efac-4855-8afb-c93af3380531", "iteration":"04f1c1fa-11ec-4e59-bb23-4c7aca353665", "created":"2021-06-10T17:58:58.959Z", "predictions":[ { "probability":0.5582016, "tagId":"05a432ea-9718-4098-b14f-5f0688149d64", "tagName":"ripe" }, { "probability":0.44179836, "tagId":"bb091037-16e5-418e-a9ea-31c6a2920f17", "tagName":"unripe" } ] }
這裡重要的部分是
predictions
陣列。它包含了預測結果,每個標籤都有一個條目,包含標籤名稱和概率。返回的概率是 0-1 的浮點數,其中 0 表示與該標籤匹配的概率為 0%,1 表示 100%。💁 圖像分類器會返回所有使用過的標籤的百分比。每個標籤都會有一個影像與該標籤匹配的概率。
JSON 被解碼後,每個標籤的概率會被發送到序列監視器。
-
在
buttonPressed
函數中,將保存到 SD 卡的程式碼替換為對classifyImage
的呼叫,或者在影像寫入後但 刪除緩衝區之前 新增該呼叫:classifyImage(buffer, length);
💁 如果你替換了保存到 SD 卡的程式碼,可以清理程式碼,移除
setupSDCard
和saveToSDCard
函數。 -
上傳並運行你的程式。將相機對準一些水果,按下 C 按鈕。你會在序列監視器中看到輸出:
Connecting to WiFi.. Connected! Image captured Image read to buffer with length 8200 ripe: 56.84% unripe: 43.16%
你將能夠看到拍攝的影像,並在 Custom Vision 的 Predictions 標籤中看到這些值。
💁 你可以在 code-classify/wio-terminal 資料夾中找到這段程式碼。
😀 你的水果品質分類器程式成功了!
免責聲明:
本文件已使用人工智能翻譯服務 Co-op Translator 進行翻譯。儘管我們致力於提供準確的翻譯,但請注意,自動翻譯可能包含錯誤或不準確之處。原始文件的母語版本應被視為權威來源。對於重要信息,建議使用專業人工翻譯。我們對因使用此翻譯而引起的任何誤解或錯誤解釋概不負責。