65 lines
1.3 KiB
Go
65 lines
1.3 KiB
Go
package face
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
// Points is a list of face landmark coordinates.
|
|
type Points []Point
|
|
|
|
// Markers returns relative marker positions for all face landmark coordinates.
|
|
func (pts Points) Markers(r Point, rows, cols float32) (m Markers) {
|
|
for _, p := range pts {
|
|
m = append(m, p.Marker(r, rows, cols))
|
|
}
|
|
|
|
return m
|
|
}
|
|
|
|
// Point represents a face landmark position.
|
|
type Point struct {
|
|
Name string `json:"name,omitempty"`
|
|
Row int `json:"x,omitempty"`
|
|
Col int `json:"y,omitempty"`
|
|
Scale int `json:"size,omitempty"`
|
|
}
|
|
|
|
// String returns the face landmark position as string.
|
|
func (p Point) String() string {
|
|
return fmt.Sprintf("%d-%d-%d", p.Row, p.Col, p.Scale)
|
|
}
|
|
|
|
// NewPoint returns new face landmark coordinates.
|
|
func NewPoint(name string, row, col, scale int) Point {
|
|
return Point{
|
|
Name: name,
|
|
Row: row,
|
|
Col: col,
|
|
Scale: scale,
|
|
}
|
|
}
|
|
|
|
// Marker returns a relative marker position for the face landmark coordinates.
|
|
func (p Point) Marker(r Point, rows, cols float32) Marker {
|
|
if rows < 1 {
|
|
rows = 1
|
|
}
|
|
|
|
if cols < 1 {
|
|
cols = 1
|
|
}
|
|
|
|
return NewMarker(
|
|
p.Name,
|
|
float32(p.Col-r.Col)/cols,
|
|
float32(p.Row-r.Row)/rows,
|
|
float32(p.Scale)/rows,
|
|
float32(p.Scale)/cols,
|
|
)
|
|
}
|
|
|
|
// TopLeft returns the top left position of the face.
|
|
func (p Point) TopLeft() (int, int) {
|
|
return p.Row - (p.Scale / 2), p.Col - (p.Scale / 2)
|
|
}
|