arcface

Documentation

Index

• func NewFactory() func() TModel
• type Input
• type Model
  • func New() *Model
  • func (m *Model) ModelName() string
  • func (m *Model) ModelVersion() string
  • func (m *Model) PostProcess(rawOutputContents [][]byte) (*Output, error)
  • func (m *Model) PreProcess(i *Input) ([]*protobuf.InferTensorContents, error)
• type Output
• type TModel

Functions

func NewFactory

func NewFactory() func() TModel

Types

type Input

type Input struct {
	Img           gocv.Mat
	FaceLandmark5 []gocv.Point2f
}

type Model

type Model struct{}

func New

func New() *Model

func (*Model) ModelName

func (m *Model) ModelName() string

func (*Model) ModelVersion

func (m *Model) ModelVersion() string

func (*Model) PostProcess

func (m *Model) PostProcess(rawOutputContents [][]byte) (*Output, error)

func (*Model) PreProcess

func (m *Model) PreProcess(i *Input) ([]*protobuf.InferTensorContents, error)

type Output

type Output struct {
	Embedding       []float32
	NormedEmbedding []float32
}

type TModel

type TModel = model.Model[*Input, *Output]