mirror of
https://github.com/dev6699/face.git
synced 2025-12-24 11:51:07 +08:00
21 lines
353 B
Go
21 lines
353 B
Go
package model
|
|
|
|
import (
|
|
"github.com/dev6699/face/protobuf"
|
|
)
|
|
|
|
type Model[I any, O any] interface {
|
|
ModelMeta
|
|
PreProcess(input I) ([]*protobuf.InferTensorContents, error)
|
|
PostProcess(rawOutputContents [][]byte) (O, error)
|
|
}
|
|
|
|
type ModelMeta interface {
|
|
ModelName() string
|
|
ModelVersion() string
|
|
}
|
|
|
|
type BoundingBox struct {
|
|
X1, Y1, X2, Y2 float64
|
|
}
|