io.github.triniwiz.fancycamera.objectdetection.Result.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of object-detection Show documentation
Fancy Android Camera ObjectDetection
The newest version!
package io.github.triniwiz.fancycamera.objectdetection

import android.graphics.Rect
import com.google.gson.annotations.SerializedName
import com.google.mlkit.vision.objects.DetectedObject
import com.google.mlkit.vision.objects.defaults.PredefinedCategory


class Result(detectedObject: DetectedObject) {
    val trackingId = detectedObject.trackingId
    val bounds = Bounds(detectedObject.boundingBox)
    val labels: Array

    init {
        val labels = mutableListOf()
        for (label in detectedObject.labels) {
            detectedObject.labels
            labels.add(Label(label))
        }
        this.labels = labels.toTypedArray()
    }

    class Label(label: DetectedObject.Label) {
        val text: Category
        val confidence: Float = label.confidence
        val index: Index

        init {
            text = when (label.text) {
                PredefinedCategory.FASHION_GOOD -> Category.FashionGood
                PredefinedCategory.FOOD -> Category.Food
                PredefinedCategory.HOME_GOOD -> Category.HomeGood
                PredefinedCategory.PLACE -> Category.Place
                PredefinedCategory.PLANT -> Category.Plant
                else -> Category.Unknown
            }
            index = when (label.index) {
                PredefinedCategory.FASHION_GOOD_INDEX -> Index.FashionGoodIndex
                PredefinedCategory.FOOD_INDEX -> Index.FoodIndex
                PredefinedCategory.HOME_GOOD_INDEX -> Index.HomeGoodIndex
                PredefinedCategory.PLACE_INDEX -> Index.PlaceIndex
                PredefinedCategory.PLANT_INDEX -> Index.PlantIndex
                else -> Index.UnknownIndex
            }
        }

        enum class Category(val category: String) {
            @SerializedName("unknown")
            Unknown("unknown"),

            @SerializedName("homeGood")
            HomeGood("homeGood"),

            @SerializedName("fashionGood")
            FashionGood("fashionGood"),

            @SerializedName("food")
            Food("food"),

            @SerializedName("place")
            Place("place"),

            @SerializedName("plant")
            Plant("plant")
        }

        enum class Index(val index: String) {
            @SerializedName("unknownIndex")
            UnknownIndex("unknownIndex"),

            @SerializedName("homeGoodIndex")
            HomeGoodIndex("homeGoodIndex"),

            @SerializedName("fashionGoodIndex")
            FashionGoodIndex("fashionGoodIndex"),

            @SerializedName("foodIndex")
            FoodIndex("foodIndex"),

            @SerializedName("placeIndex")
            PlaceIndex("placeIndex"),

            @SerializedName("plantIndex")
            PlantIndex("plantIndex")
        }

    }

    class Bounds(rect: Rect) {
        val x: Int = rect.left

        val y: Int = rect.top

        val width: Int = rect.width()

        val height: Int = rect.height()
    }
}