gofaces

A sample project to run a tiny yolo v2 with face detection in Go

APACHE-2.0 License

Stars
12

gofaces

This repository is an implementation of this article.

It is a toy to run face detection on a picture with a neural network.

The neural network is based on Tiny YOLO v2 and is encoded in the onnx format. It decocded with onnx-go and executed by Gorgonia.

Installation an usage

Requirements

  • Go with modules support, 1.12+ recommended.
  • Git LFS. It's needed in order to get the file model.onnx when cloning the repo. If you dont have this, running will fail with an error: proto: cant skip unknown wire type 6

Usage

git clone https://github.com/owulveryck/gofaces
cd gofaces
git lfs install
git lfs fetch
git lfs checkout model/model.onnx
cd cmd
go run main.go -h