genderage

Documentation

Index

• func NewFactory() func() ModelT
• 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 ModelT
• type Output

Functions

func NewFactory

func NewFactory() func() ModelT

Types

type Input

type Input struct {
	Img         gocv.Mat
	BoundingBox model.BoundingBox
}

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 ModelT

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

type Output

type Output struct {
	Age int
	// Gender: female=0, male=1
	Gender int
}