Add example

This commit is contained in:
Kagami Hiiragi
2018-07-31 01:54:12 +03:00
parent 931d5f4771
commit 7d71b46f63
2 changed files with 34 additions and 1 deletions

33
example_test.go Normal file
View 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)
}

View File

@@ -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)
} }
} }