https://github.com/jdeng/goface

Face Detector based on MTCNN, tensorflow and golang
https://github.com/jdeng/goface

face-detector golang mtcnn tensorflow

Last synced: 21 days ago
JSON representation

Face Detector based on MTCNN, tensorflow and golang

• Host: GitHub
• URL: https://github.com/jdeng/goface
• Owner: jdeng
• License: mit
• Created: 2018-01-23T19:18:17.000Z (almost 7 years ago)
• Default Branch: master
• Last Pushed: 2022-08-04T17:03:41.000Z (over 2 years ago)
• Last Synced: 2024-12-18T22:02:00.552Z (28 days ago)
• Topics: face-detector, golang, mtcnn, tensorflow
• Language: Go
• Size: 1.75 MB
• Stars: 119
• Watchers: 9
• Forks: 26
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# 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.