drawer

Documentation

Index

• Constants
• func EulrToAxisPoints(yaw float64, pitch float64, roll float64, distance float64) [3][2]float64
• type Drawer
  • func New(options ...Option) *Drawer
  • func (d *Drawer) Draw(img image.Image, faces []face.FaceInfo) image.Image
  • func (d *Drawer) DrawHeadPose(img image.Image, pose face.HeadPose, rect common.Rectangle) image.Image
  • func (d *Drawer) DrawLandmark(img image.Image, points []common.Point) image.Image
• type Option
  • func WithBorderColor(color string) Option
  • func WithBorderStrokeWidth(w float64) Option
  • func WithFont(font *common.Font) Option
  • func WithInvalidBorderColor(color string) Option
  • func WithKeypointColor(color string) Option
  • func WithKeypointRadius(r float64) Option
  • func WithKeypointStrokeWidth(w float64) Option
  • func WithLabelColor(color string) Option
  • func WithMaskColor(color string) Option

Constants

const (
	// DefaultBorderColor default drawer border color
	DefaultBorderColor = common.Green
	// DefaultKeypointColor default drawer keypoint color
	DefaultKeypointColor = common.Pink
	// DefaultBorderStrokeWidth default drawer border stroke width
	DefaultBorderStrokeWidth = 3
	// DefaultKeypointRadius default drawer keypoint radius
	DefaultKeypointRadius = 2
	// DefaultKeypointStrokeWidth default drawer keypoint stroke width
	DefaultKeypointStrokeWidth = 2
	// DefaultInvalidBorderColor default drawer invalid border color
	DefaultInvalidBorderColor = common.Red
	// DefaultLabelColor default label color
	DefaultLabelColor = common.White
)

Functions

func EulrToAxisPoints

func EulrToAxisPoints(yaw float64, pitch float64, roll float64, distance float64) [3][2]float64

EulrToAxisPoints returns a transformation roll/pitch/yaw to Axis in 2d intrinsic rotations around the (local) axis Y, X, Z, in that order.

Types

type Drawer

type Drawer struct {
	// BorderColor represents face rect border color
	BorderColor string
	// KeypointColor represents keypoint color
	KeypointColor string
	// BorderStrokeWidth represents face rect stroke width
	BorderStrokeWidth float64
	// KeypointRadius represents keypoints circle radius
	KeypointRadius float64
	// KeypointStrokeWidth represents keypoints stroke width
	KeypointStrokeWidth float64
	// MaskColor represents border color which mask is true
	MaskColor string
	// InvalidBorderColor
	InvalidBorderColor string
	// LabelColor string
	LabelColor string
	// Font
	Font *common.Font
}

Drawer represents a face drawer

func New

func New(options ...Option) *Drawer

New returns a new Drawer

func (*Drawer) Draw

func (d *Drawer) Draw(img image.Image, faces []face.FaceInfo) image.Image

Draw draw faces

func (*Drawer) DrawHeadPose

func (d *Drawer) DrawHeadPose(img image.Image, pose face.HeadPose, rect common.Rectangle) image.Image

DrawHeadPose draw head pose

func (*Drawer) DrawLandmark

func (d *Drawer) DrawLandmark(img image.Image, points []common.Point) image.Image

DrawLandmark draw landmark

type Option

type Option interface {
	// contains filtered or unexported methods
}

Option represents Drawer option interface

func WithBorderColor

func WithBorderColor(color string) Option

WithBorderColor set Drawer BorderColor

func WithBorderStrokeWidth

func WithBorderStrokeWidth(w float64) Option

WithBorderStrokeWidth set Drawer BorderStrokeWidth

func WithFont

func WithFont(font *common.Font) Option

WithFont set Drawer Font

func WithInvalidBorderColor

func WithInvalidBorderColor(color string) Option

WithInvalidBorderColor set Drawer InvalidBorderColor

func WithKeypointColor

func WithKeypointColor(color string) Option

WithKeypointColor set Drawer KeypointColor

func WithKeypointRadius

func WithKeypointRadius(r float64) Option

WithKeypointRadius set Drawer KeypointRadius

func WithKeypointStrokeWidth

func WithKeypointStrokeWidth(w float64) Option

WithKeypointStrokeWidth set Drawer KeypointStrokeWidth

func WithLabelColor

func WithLabelColor(color string) Option

WithLabelColor set Drawer LabelColor

func WithMaskColor

func WithMaskColor(color string) Option

WithMaskColor set Drawer MaskColor