mirror of
https://github.com/dev6699/face.git
synced 2025-09-26 21:16:00 +08:00
17 lines
300 B
Go
17 lines
300 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
|
|
}
|