Instalação Darknet YOLOv3 - Linux Mint
Dependências:
- OpenCV
- interpretador de C
- Nvidia CUDA toolkit
Antes de mais nada, verifique se sua máquina tem um interpretador de C(GCC) instalado. Caso não esteja instalado, para prosseguir com a instalação você pode instalar com o seguinte comando no terminal:
sudo apt-get install libc-dev
Agora, devemos seguir com a instalação do OpenCV. Instale com o seguinte comando:
sudo apt-get install libopencv-dev
Para finalizar a instalação de dependências, deve ser instalado o Nvidia CUDA toolkit usando o comando:
sudo apt-get install nvidia-cuda-toolkit
Após as dependências estarem instaladas, e ainda no terminal, navegue até o diretório no qual deseja instalar o Darknet
para executar os comandos. Execute os seguintes comandos, um após a excução completa do anterior:
git clone https://github.com/pjreddie/darknet.git
cd darknet
make
Ao final da intalação, é necessário sinalizar ao YOLOv3
que você irá utilizar ele. Para isso, entre na pasta do darknet e procure o arquivo Makefile
. Abra o arquivo e altere na terceira linha OPENCV=0
para OPENCV=1
, altere na primera linha GPU=0
para GPU=1
salve e volte para o diretório darknet no terminal para rebuildar a instalação com o comando:
make
Para testar a detecção de imagens estáticas, será necessário baixar o arquivo yolov3.weights
através do link e colar ele na própria pasta darknet. Após, a aplicação já pode ser testada através do comando:
./darknet detector test cfg/coco.data cfg/yolov3.cfg yolov3.weights data/dog.jp
O teste gera um arquivo chamado predictions.png com as detecções.
O YOLOv3, por padrão, após analisar uma imagem, salva a imagem na pasta darknet. Ao utilizarmos o OpenCV isso é feito em memória (o resultado é exibido e não fica salvo na pasta).