estimator

Documentation

Overview

Package estimator pose estimator

Index

• func ExtractKeypoints(d Estimator, img *common.Image, rect common.Rectangle) ([]common.Keypoint, error)
• type Estimator
• type MoveNet
  • func NewMoveNet(modelType MoveNetType) *MoveNet
  • func (d *MoveNet) Destroy()
  • func (d *MoveNet) ExtractKeypoints(img *common.Image, rect common.Rectangle) ([]common.Keypoint, error)
  • func (d *MoveNet) LoadModel(modelPath string) error
  • func (d *MoveNet) Pointer() unsafe.Pointer
• type MoveNetType
• type Ultralight
  • func NewUltralight() *Ultralight
  • func (d *Ultralight) Destroy()
  • func (d *Ultralight) ExtractKeypoints(img *common.Image, rect common.Rectangle) ([]common.Keypoint, error)
  • func (d *Ultralight) LoadModel(modelPath string) error
  • func (d *Ultralight) Pointer() unsafe.Pointer

Functions

func ExtractKeypoints

func ExtractKeypoints(d Estimator, img *common.Image, rect common.Rectangle) ([]common.Keypoint, error)

ExtractKeypoints detect pose keypoints using estimator

Types

type Estimator

type Estimator interface {
	common.Estimator
	ExtractKeypoints(img *common.Image, rect common.Rectangle) ([]common.Keypoint, error)
}

Estimator represents deteter interface

type MoveNet

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

MoveNet represents movenet estimator

func NewMoveNet

func NewMoveNet(modelType MoveNetType) *MoveNet

NewMoveNet returns a new MoveNet

func (*MoveNet) Destroy

func (d *MoveNet) Destroy()

Destroy free detecter

func (*MoveNet) ExtractKeypoints

func (d *MoveNet) ExtractKeypoints(img *common.Image, rect common.Rectangle) ([]common.Keypoint, error)

ExtractKeypoints implement Detecter interface

func (*MoveNet) LoadModel

func (d *MoveNet) LoadModel(modelPath string) error

LoadModel load model for detecter

func (*MoveNet) Pointer

func (d *MoveNet) Pointer() unsafe.Pointer

Pointer implement Estimator interface

type MoveNetType

type MoveNetType = int

MoveNetType (lightning/thunder)

const (
	// MoveNetType_Lightning lightning model
	MoveNetType_Lightning MoveNetType = 0
	// MoveNetType_Thunder thunder model
	MoveNetType_Thunder MoveNetType = 1
)

type Ultralight

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

Ultralight represents utralight detecter

func NewUltralight

func NewUltralight() *Ultralight

NewUltralight returns a new Utralight

func (*Ultralight) Destroy

func (d *Ultralight) Destroy()

Destroy free detecter

func (*Ultralight) ExtractKeypoints

func (d *Ultralight) ExtractKeypoints(img *common.Image, rect common.Rectangle) ([]common.Keypoint, error)

ExtractKeypoints implement Detecter interface

func (*Ultralight) LoadModel

func (d *Ultralight) LoadModel(modelPath string) error

LoadModel load model for detecter

func (*Ultralight) Pointer

func (d *Ultralight) Pointer() unsafe.Pointer

Pointer implement Estimator interface