Documentation ¶
Overview ¶
Package face implements face recognition for Go using dlib, a popular machine learning toolkit.
Index ¶
- func SquaredEuclideanDistance(d1 Descriptor, d2 Descriptor) (sum float64)
- type Descriptor
- type Face
- type ImageLoadError
- type Recognizer
- func (rec *Recognizer) Classify(testSample Descriptor) int
- func (rec *Recognizer) ClassifyThreshold(testSample Descriptor, tolerance float32) int
- func (rec *Recognizer) Close()
- func (rec *Recognizer) Deserialize(type_ int, buf []byte) error
- func (rec *Recognizer) Recognize(imgData []byte) (faces []Face, err error)
- func (rec *Recognizer) RecognizeCNN(imgData []byte) (faces []Face, err error)
- func (rec *Recognizer) RecognizeFile(imgPath string) (faces []Face, err error)
- func (rec *Recognizer) RecognizeFileCNN(imgPath string) (faces []Face, err error)
- func (rec *Recognizer) RecognizeSingle(imgData []byte) (face *Face, err error)
- func (rec *Recognizer) RecognizeSingleCNN(imgData []byte) (face *Face, err error)
- func (rec *Recognizer) RecognizeSingleFile(imgPath string) (face *Face, err error)
- func (rec *Recognizer) RecognizeSingleFileCNN(imgPath string) (face *Face, err error)
- func (rec *Recognizer) SetSamples(samples []Descriptor, cats []int32)
- type SerializationError
- type UnknownError
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func SquaredEuclideanDistance ¶
func SquaredEuclideanDistance(d1 Descriptor, d2 Descriptor) (sum float64)
Types ¶
type Face ¶
type Face struct { Rectangle image.Rectangle Descriptor Descriptor Shapes []image.Point }
Face holds coordinates and descriptor of the human face.
func New ¶
func New(r image.Rectangle, d Descriptor) Face
New creates new face with the provided parameters.
func NewWithShape ¶
type ImageLoadError ¶
type ImageLoadError string
An ImageLoadError is returned when provided image file is corrupted.
func (ImageLoadError) Error ¶
func (e ImageLoadError) Error() string
type Recognizer ¶
type Recognizer struct {
// contains filtered or unexported fields
}
A Recognizer creates face descriptors for provided images and classifies them into categories.
func NewRecognizer ¶
func NewRecognizer() (rec *Recognizer, err error)
NewRecognizer returns a new recognizer interface. modelDir points to directory with shape_predictor_5_face_landmarks.dat and dlib_face_recognition_resnet_model_v1.dat files.
func NewRecognizerWithConfig ¶
func NewRecognizerWithConfig(size int, padding float32, jittering int) (rec *Recognizer, err error)
func (*Recognizer) Classify ¶
func (rec *Recognizer) Classify(testSample Descriptor) int
Classify returns class ID for the given descriptor. Negative index is returned if no match. Thread-safe.
func (*Recognizer) ClassifyThreshold ¶
func (rec *Recognizer) ClassifyThreshold(testSample Descriptor, tolerance float32) int
Same as Classify but allows to specify max distance between faces to consider it a match. Start with 0.6 if not sure.
func (*Recognizer) Close ¶
func (rec *Recognizer) Close()
Close frees resources taken by the Recognizer. Safe to call multiple times. Don't use Recognizer after close call.
func (*Recognizer) Deserialize ¶
func (rec *Recognizer) Deserialize(type_ int, buf []byte) error
func (*Recognizer) Recognize ¶
func (rec *Recognizer) Recognize(imgData []byte) (faces []Face, err error)
Recognize returns all faces found on the provided image, sorted from left to right. Empty list is returned if there are no faces, error is returned if there was some error while decoding/processing image. Only JPEG format is currently supported. Thread-safe.
func (*Recognizer) RecognizeCNN ¶
func (rec *Recognizer) RecognizeCNN(imgData []byte) (faces []Face, err error)
func (*Recognizer) RecognizeFile ¶
func (rec *Recognizer) RecognizeFile(imgPath string) (faces []Face, err error)
Same as Recognize but accepts image path instead.
func (*Recognizer) RecognizeFileCNN ¶
func (rec *Recognizer) RecognizeFileCNN(imgPath string) (faces []Face, err error)
func (*Recognizer) RecognizeSingle ¶
func (rec *Recognizer) RecognizeSingle(imgData []byte) (face *Face, err error)
RecognizeSingle returns face if it's the only face on the image or nil otherwise. Only JPEG format is currently supported. Thread-safe.
func (*Recognizer) RecognizeSingleCNN ¶
func (rec *Recognizer) RecognizeSingleCNN(imgData []byte) (face *Face, err error)
func (*Recognizer) RecognizeSingleFile ¶
func (rec *Recognizer) RecognizeSingleFile(imgPath string) (face *Face, err error)
Same as RecognizeSingle but accepts image path instead.
func (*Recognizer) RecognizeSingleFileCNN ¶
func (rec *Recognizer) RecognizeSingleFileCNN(imgPath string) (face *Face, err error)
func (*Recognizer) SetSamples ¶
func (rec *Recognizer) SetSamples(samples []Descriptor, cats []int32)
SetSamples sets known descriptors so you can classify the new ones. Thread-safe.
type SerializationError ¶
type SerializationError string
An SerializationError is returned when provided model is corrupted.
func (SerializationError) Error ¶
func (e SerializationError) Error() string
type UnknownError ¶
type UnknownError string
An UnknownError represents some nonclassified error.
func (UnknownError) Error ¶
func (e UnknownError) Error() string