Face detection and recognition using golang
go-recognizer require go-face to compile. go-face need to have dlib (>= 19.10) and libjpeg development packages installed.
Latest versions of Ubuntu and Debian provide suitable dlib package so just run:
# Ubuntu
sudo apt-get install libdlib-dev libblas-dev libatlas-base-dev liblapack-dev libjpeg-turbo8-dev
# Debian
sudo apt-get install libdlib-dev libblas-dev libatlas-base-dev liblapack-dev libjpeg62-turbo-dev
Make sure you have Homebrew installed.
brew install dlib
Make sure you have MSYS2 installed.
MSYS2 MSYS
shell from Start menupacman -Syu
and if it asks you to close the shell do thatpacman -Syu
againpacman -S mingw-w64-x86_64-gcc mingw-w64-x86_64-dlib
set MSYS2_PATH_TYPE=inherit
line in msys2_shell.cmd
located in MSYS2
installation folderpacman -S mingw-w64-x86_64-go git
MSYS2 MinGW 64-bit
shell from Start menu to compile and use go-faceTry to install dlib/libjpeg with package manager of your distribution or compile from sources. Note that go-face won't work with old packages of dlib such as libdlib18. Alternatively create issue with the name of your system and someone might help you with the installation process.
To use go-recognizer in your Go code:
import "github.com/leandroveronezi/go-recognizer"
To install go-recognizer in your $GOPATH:
go get github.com/leandroveronezi/go-recognizer
Currently shape_predictor_5_face_landmarks.dat
, mmod_human_face_detector.dat
and
dlib_face_recognition_resnet_model_v1.dat
are required. You may download them
from dlib-models repo:
mkdir models && cd models
wget https://github.com/davisking/dlib-models/raw/master/shape_predictor_5_face_landmarks.dat.bz2
bunzip2 shape_predictor_5_face_landmarks.dat.bz2
wget https://github.com/davisking/dlib-models/raw/master/dlib_face_recognition_resnet_model_v1.dat.bz2
bunzip2 dlib_face_recognition_resnet_model_v1.dat.bz2
wget https://github.com/davisking/dlib-models/raw/master/mmod_human_face_detector.dat.bz2
bunzip2 mmod_human_face_detector.dat.bz2
package main
import (
"fmt"
"github.com/leandroveronezi/go-recognizer"
"path/filepath"
)
const fotosDir = "fotos"
const dataDir = "models"
func main() {
rec := recognizer.Recognizer{}
err := rec.Init(dataDir)
if err != nil {
fmt.Println(err)
return
}
rec.Tolerance = 0.4
rec.UseGray = true
rec.UseCNN = false
defer rec.Close()
faces, err := rec.RecognizeMultiples(filepath.Join(fotosDir, "elenco3.jpg"))
if err != nil {
fmt.Println(err)
return
}
img, err := rec.DrawFaces2(filepath.Join(fotosDir, "elenco3.jpg"), faces)
if err != nil {
fmt.Println(err)
return
}
rec.SaveImage("faces2.jpeg", img)
}
package main
import (
"fmt"
"github.com/leandroveronezi/go-recognizer"
"path/filepath"
)
const fotosDir = "fotos"
const dataDir = "models"
func addFile(rec *recognizer.Recognizer, Path, Id string) {
err := rec.AddImageToDataset(Path, Id)
if err != nil {
fmt.Println(err)
return
}
}
func main() {
rec := recognizer.Recognizer{}
err := rec.Init(dataDir)
if err != nil {
fmt.Println(err)
return
}
rec.Tolerance = 0.4
rec.UseGray = true
rec.UseCNN = false
defer rec.Close()
addFile(&rec, filepath.Join(fotosDir, "amy.jpg"), "Amy")
addFile(&rec, filepath.Join(fotosDir, "bernadette.jpg"), "Bernadette")
addFile(&rec, filepath.Join(fotosDir, "howard.jpg"), "Howard")
addFile(&rec, filepath.Join(fotosDir, "penny.jpg"), "Penny")
addFile(&rec, filepath.Join(fotosDir, "raj.jpg"), "Raj")
addFile(&rec, filepath.Join(fotosDir, "sheldon.jpg"), "Sheldon")
addFile(&rec, filepath.Join(fotosDir, "leonard.jpg"), "Leonard")
rec.SetSamples()
faces, err := rec.ClassifyMultiples(filepath.Join(fotosDir, "elenco3.jpg"))
if err != nil {
fmt.Println(err)
return
}
img, err := rec.DrawFaces(filepath.Join(fotosDir, "elenco3.jpg"), faces)
if err != nil {
fmt.Println(err)
return
}
rec.SaveImage("faces.jpg", img)
}