com.github.charleslzq.baiduface.client.io.RequestDTO.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of baidu-face-server Show documentation

baidu face server, make request to baidu face open api for client

The newest version!

package com.github.charleslzq.baiduface.client.io

import com.baidu.aip.face.FaceVerifyRequest
import com.baidu.aip.face.MatchRequest

data class Image(
        val type: Type = Type.BASE64,
        val data: String = ""
) {
    enum class Type {
        BASE64,
        URL,
        FACE_TOKEN
    }
}

enum class FaceField(val keyword: String) {
    AGE("age"),
    BEAUTY("beauty"),
    EXPRESSION("expression"),
    FACE_SHAPE("faceshape"),
    GENDER("gender"),
    GLASSES("glasses"),
    LANDMARK("landmark"),
    RACE("race"),
    QUALITY("quality"),
    FACE_TYPE("facetype"),
    PARSING("parsing");
}

enum class FaceSource {
    LIVE,
    IDCARD,
    WATERMARK,
    CERT
}

data class UserMeta(
        val groupId: String,
        val userId: String,
        val userInfo: String? = null
)

enum class QualityControl {
    NONE,
    LOW,
    NORMAL,
    HIGH
}

enum class LivenessControl {
    NONE,
    LOW,
    NORMAL,
    HIGH
}

data class MatchReq(
        val image: Image = Image(),
        val source: FaceSource = FaceSource.LIVE,
        val quality: QualityControl = QualityControl.NONE,
        val liveness: LivenessControl = LivenessControl.NONE
) {
    fun toMatchRequest() = MatchRequest(
            image.data,
            image.type.name,
            source.name,
            quality.name,
            liveness.name
    )
}

data class FaceVerifyReq(
        val image: Image = Image(),
        val complete: Boolean = false,
        val fields: List = emptyList()
) {
    fun toFaceVerifyRequest() = FaceVerifyRequest(
            image.data,
            image.type.name,
            (if (complete) FaceField.values().toList() else fields).string()
    )
}