yoloface

YOLOFace

Detecção de rosto baseada em aprendizado profundo usando o algoritmo YOLOv3

Começando

O YOLOv3 (You Only Look Once) é um algoritmo de detecção de objetos em tempo real de última geração. O modelo publicado reconhece 80 objetos diferentes em imagens e vídeos. Para obter mais detalhes, você pode consultar este artigo.

Arquitetura do YOLOv3

Crédito: Ayoosh Kathuria

Redes Neurais Profundas OpenCV (módulo dnn)

O módulo OpenCV dnn suporta a execução de inferência em modelos de aprendizado profundo pré-treinados de estruturas populares como TensorFlow, Torch, Darknet e Caffe.

Pré-requisitos

• Fluxo tensor
• opencv-python
• opencv-contrib-python
• Entorpecido
• Keras
• Matplotlib
• Travesseiro

O desenvolvimento deste projeto será isolado em ambiente virtual Python. Isso nos permite experimentar diferentes versões de dependências.

Existem muitas maneiras de instalar virtual environment (virtualenv) , consulte o guia Python Virtual Environments: A Primer para diferentes plataformas, mas aqui estão algumas:

• Para Ubuntu

$ pip install virtualenv

• Para Mac

$ pip install --upgrade virtualenv

Crie um ambiente virtual Python 3.6 para este projeto e ative o virtualenv:

$ virtualenv -p python3.6 yoloface
$ source ./yoloface/bin/activate

A seguir, instale as dependências deste projeto:

$ pip install -r requirements.txt

Uso

• Clonar este repositório

$ git clone https://github.com/sthanhng/yoloface

• Para detecção de rosto, você deve baixar o arquivo de pesos YOLOv3 pré-treinado que foi treinado no conjunto de dados WIDER FACE: A Face Detection Benchmark deste link e colocá-lo no diretório model-weights/ .

• Execute o seguinte comando:

entrada de imagem

$ python yoloface.py --image samples/outside_000001.jpg --output-dir outputs/

entrada de vídeo

$ python yoloface.py --video samples/subway.mp4 --output-dir outputs/

Webcam

$ python yoloface.py --src 1 --output-dir outputs/

Exemplos de resultados

Licença

Este projeto está licenciado sob a licença MIT - consulte o arquivo LICENSE.md para obter mais detalhes.

Referências