快捷導航

基于Google ML模型開發(fā)Android物體檢測應(yīng)用

更新時間：2024年07月27日 15:12:17 作者：青蛙娃娃

ML Kit是Google提供的機器學習SDK,包含了一系列預訓練模型,可以在Android和iOS應(yīng)用中快速添加機器學習功能,本項目基于Google ML模型開發(fā)Android物體檢測應(yīng)用,首先對圖像中的物體進行分類檢測,獲取分類物體的位置區(qū)域,然后結(jié)合圖像標記,逐個獲取單個物體的標簽

ML Kit

ML Kit是Google提供的機器學習SDK，包含了一系列預訓練模型，可以在Android和iOS應(yīng)用中快速添加機器學習功能。ML Kit工具包將谷歌在機器學習方面的專業(yè)知識帶給了普通的移動應(yīng)用開發(fā)者，可以借助ML Kit在幾行代碼中實現(xiàn)所需的功能。

ML Kit的功能非常全面且易于使用，提供了基于API的現(xiàn)有解決方案，如文字識別、條碼識別、圖像標簽、人臉檢測、對象檢測等。同時，還提供了便利的API，允許在移動應(yīng)用中使用自定義的TensorFlow Lite模型。

ML Kit針對移動設(shè)備進行了優(yōu)化，使得機器學習套件的處理過程可以在設(shè)備上進行，保證了快速性，并解鎖了處理相機輸入等實時用例的能力。即使在離線狀態(tài)下，也可以使用這項功能來處理需要保留在設(shè)備上的圖片和文本。

項目描述

ML Kit的默認模型在物體檢測方面表現(xiàn)非常出色。當你將圖像傳遞給物體檢測器時，它會返回一個物體列表，其中包含邊界框，這些邊界框可用于確定圖像中物體可能的位置。

本項目首先對圖像中的物體進行分類檢測，獲取分類物體的位置區(qū)域，然后結(jié)合圖像標記，逐個獲取單個物體的標簽。

開發(fā)工具

Android Studio Koala
Kotlin
Gradle 8.7

代碼設(shè)計

界面設(shè)計

界面上方是一個ImageView，用于呈現(xiàn)示例圖片；用戶點擊中間按鈕，調(diào)用物體檢測處理代碼；待處理完畢，將結(jié)果展示在下方的 TextView 中。

物體檢測處理

ML Kit物體檢測器提供多種物體檢測方法，這些方法由ObjectDetectorOptions對象控制。該檢測器是一個強大的API，不僅可以檢測物體，還能在視頻流中跟蹤它們，實現(xiàn)逐幀跟蹤功能。

創(chuàng)建物體檢測器，將Bitmap轉(zhuǎn)換為InputImage，并使用物體檢測器進行處理。這將在成功時返回檢測到的物體列表，或在失敗時返回異常物體。

val objectDetector = ObjectDetection.getClient(options)
            var image = InputImage.fromBitmap(bitmap!!, 0)
            txtOutput.text = ""
            objectDetector.process(image)
                    .addOnSuccessListener { detectedObjects -&gt;
                        // 任務(wù)執(zhí)行成功
                        getLabels(bitmap, detectedObjects, txtOutput)
                        bitmap?.apply{
                            img.setImageBitmap(drawWithRectangle(detectedObjects))
                        }

                    }
                    .addOnFailureListener { e -&gt;
                        // 任務(wù)執(zhí)行失敗
                        Log.e("MainActivity", e.toString())
                    }

用戶點擊按鈕后，調(diào)用物體檢測器以獲取圖像中物體的邊界框。然后，使用這些邊界框裁剪圖像，得到定義的子圖像，并將其傳遞給圖像標記器。

標記物體

使用已有的邊界框創(chuàng)建新的臨時圖像（croppedBitmap），將其傳遞給圖像標記器，并返回結(jié)果。重復此操作以處理每個邊界框（即每個物體），以獲取每個檢測到物體的詳細標簽！

val labeler =
        ImageLabeling.getClient(ImageLabelerOptions.DEFAULT_OPTIONS)
    for(obj in objects) {
        val bounds = obj.boundingBox
        val croppedBitmap = Bitmap.createBitmap(
            bitmap,
            bounds.left,
            bounds.top,
            bounds.width(),
            bounds.height()
        )
        var image = InputImage.fromBitmap(croppedBitmap!!, 0)
        labeler.process(image)
            .addOnSuccessListener { labels ->
                // 任務(wù)執(zhí)行成功
                var labelText = ""
                if(labels.isNotEmpty()) {
                    labelText = txtOutput.text.toString()
                    for (thisLabel in labels){
                        labelText += thisLabel.text + " , "
                    }
                    labelText += "\n"
                } else {
                    labelText = "Not found." + "\n"
                }
                txtOutput.text = labelText.toString()
            }
    }

該代碼遍歷每個檢測到的物體，使用邊界框創(chuàng)建名為croppedBitmap的新位圖。接下來，它使用一個設(shè)置了默認選項的圖像標記器（稱為labeler）處理新圖像。處理成功后，從標簽中獲取多個結(jié)果，并將這些標簽寫入逗號分隔的字符串，最終呈現(xiàn)在TextView中。

項目演示

Android AI應(yīng)用開發(fā)：物體檢測_ML_02

到此這篇關(guān)于基于Google ML模型開發(fā)Android物體檢測應(yīng)用的文章就介紹到這了,更多相關(guān)Android AI應(yīng)用開發(fā)：物體檢測內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！