YOLOv3 Object Detection with OpenCV

YOLOv3-Detecção de objeto com OpenCV

Este projeto implementa um classificador de detecção de objetos de imagem e vídeo usando modelos yolov3 pré-treinados. Os modelos yolov3 são retirados do artigo oficial do yolov3, lançado em 2018. A implementação do yolov3 é da darknet. Além disso, este projeto implementa uma opção para realizar classificação em tempo real utilizando a webcam.

Como usar?

1. Clonar o repositório

 git clone https://github.com/iArunava/YOLOv3-Object-Detection-with-OpenCV.git

2. Mover para o diretório

 cd YOLOv3-Object-Detection-with-OpenCV

3. Para inferir sobre uma imagem armazenada em sua máquina local

 python3 yolo.py --image-path='/path/to/image/'

4. Para inferir sobre um vídeo armazenado em sua máquina local

 python3 yolo.py --video-path='/path/to/video/'

5. Para inferir em tempo real pela webcam

 python3 yolo.py

Nota: Isso funciona considerando que você tem os weights e os arquivos config no diretório yolov3-coco.
Se os arquivos estiverem localizados em outro lugar, mencione o caminho ao chamar yolov3.py . Para mais detalhes

 yolo.py --help

Inferência em imagens

Inferência em Vídeo

Clique na imagem para reproduzir o vídeo no YouTube

Inferência em tempo real

Clique na imagem para reproduzir o vídeo no YouTube

Referências

1. Detecção de objetos PyImageSearch YOLOv3 com OpenCV Blog

Licença

O código neste projeto é distribuído sob a licença MIT.