yoloface

Documentation

Index

• func NewFactory(faceDetectorScore float32, iouThreshold float64) func() ModelT
• type Detection
• type Input
• type Model
  • func New(faceDetectorScore float32, iouThreshold float64) *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
• type Resolution

Functions

func NewFactory

func NewFactory(faceDetectorScore float32, iouThreshold float64) func() ModelT

Types

type Detection

type Detection struct {
	BoundingBox   model.BoundingBox
	FaceLandmark5 []gocv.Point2f
	Confidence    float32
}

type Input

type Input struct {
	Img gocv.Mat
}

type Model

type Model struct {
	// contains filtered or unexported fields
}

func New

func New(faceDetectorScore float32, iouThreshold float64) *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 {
	Detections []Detection
}

type Resolution

type Resolution struct {
	Width  uint
	Height uint
}