mirror of
https://github.com/Kagami/go-face.git
synced 2025-10-09 01:20:06 +08:00
Add example
This commit is contained in:
33
example_test.go
Normal file
33
example_test.go
Normal file
@@ -0,0 +1,33 @@
|
|||||||
|
package face
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Path to directory with models.
|
||||||
|
modelDir = "testdata"
|
||||||
|
|
||||||
|
// Test image with 10 faces.
|
||||||
|
testImageTenFaces = "testdata/pristin.jpg"
|
||||||
|
)
|
||||||
|
|
||||||
|
// This example shows the basic usage of the package: create an
|
||||||
|
// recognizer, recognize faces, classify them using few known ones.
|
||||||
|
func Example() {
|
||||||
|
rec, err := NewRecognizer(modelDir)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Can't init face recognizer: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
faces, err := rec.RecognizeFile(testImageTenFaces)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("Can't get faces: %v", err)
|
||||||
|
}
|
||||||
|
numFaces := len(faces)
|
||||||
|
if numFaces != 10 {
|
||||||
|
log.Fatalf("Wrong number of faces: %d", numFaces)
|
||||||
|
}
|
||||||
|
fmt.Printf("Faces on %s: %v", testImageTenFaces, faces)
|
||||||
|
}
|
@@ -140,7 +140,7 @@ func TestNumFaces(t *testing.T) {
|
|||||||
t.Fatalf("Can't get faces: %v", err)
|
t.Fatalf("Can't get faces: %v", err)
|
||||||
}
|
}
|
||||||
numFaces := len(faces)
|
numFaces := len(faces)
|
||||||
if err != nil || numFaces != 10 {
|
if numFaces != 10 {
|
||||||
t.Fatalf("Wrong number of faces: %d", numFaces)
|
t.Fatalf("Wrong number of faces: %d", numFaces)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user