diff --git a/example_test.go b/example_test.go new file mode 100644 index 0000000..bd9f5b7 --- /dev/null +++ b/example_test.go @@ -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) +} diff --git a/face_test.go b/face_test.go index e71a3c9..52715f6 100644 --- a/face_test.go +++ b/face_test.go @@ -140,7 +140,7 @@ func TestNumFaces(t *testing.T) { t.Fatalf("Can't get faces: %v", err) } numFaces := len(faces) - if err != nil || numFaces != 10 { + if numFaces != 10 { t.Fatalf("Wrong number of faces: %d", numFaces) } }