Skip to content
/ goface Public

Face Detector based on MTCNN, tensorflow and golang

License

Notifications You must be signed in to change notification settings

jdeng/goface

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

goface

Face detector/embeddings based on MTCNN, tensorflow and golang

Implementation based on https://github.com/davidsandberg/facenet . Tensorflow (1.4.1) and the golang binding are required.

Model file cmd/mtcnn.pb is converted from facenet too (see scripts/convert.py. You will need to add facenet/src to PYTHONPATH to use it). You may need to regenerate the model file for a different version of tensorflow.

The facenet protobuf model file is available for download (see instructions from facenet).

Usage

	// detection
	bs, err := ioutil.ReadFile(*imgFile)
	img, err := goface.TensorFromJpeg(bs)
	det, err := goface.NewMtcnnDetector("mtcnn.pb")
	bbox, err := det.DetectFaces(img) //[][]float32, i.e., [x1,y1,x2,y2],...

	// embeddings
	mean, std := goface.MeanStd(img)
	wimg, err := goface.PrewhitenImage(img, mean, std)
	fn, err := goface.NewFacenet("facenet.pb")
	emb, err := fn.Embedding(wimg)

See cmd/detect.go. Use go build to build the binary and run with --help.

Notes

  • Not exactly the same (e.g., nms/padding is depending on tensorflow implementation).
  • Not fully tested. Performance could a little bit worse.
  • Face landmark support not implemented.

About

Face Detector based on MTCNN, tensorflow and golang

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published