Usando google colab para treinar um modelo darknet

Configurando seu drive

Como vai ficar a organização das pastas.

Vai ter uma pasta chamda treinamento. Destro dessa pasta vai ter, uma pasta para salvar os treinamentos, uma para salvar a darknet e uma o opencv.

Imagem drive

Compilando opencv e darknet

Esse passo demora bastante, mas precisamos fazer somente uma vez. Aqui vamos compilar e instalar o opencv e darknet em nosso drive pessoal.

Criando um projeto no colab

Site colab.

Abra o site do colab e aperte em "novo notebook". Com isso vai aparecer uma tela parecida com essa:

Inicio notebook

Coloque essa comando na primeira celula. Esse comando serve para o colab montar seu drive na execução dele.

from google.colab import drive
drive.mount('/content/drive')

Agora vamos fazer a instalação do opencv. Para isso vou passar os codigos de instalação aqui, mas caso queira saber o passo da instalação só ver esse artigo da natalia.

!mkdir /content/drive/MyDrive/treinamento && mkdir /content/drive/MyDrive/treinamento/opencv 
!sudo apt-get update
!sudo apt-get install build-essential cmake unzip pkg-config
!sudo apt-get install libjpeg-dev libpng-dev libtiff-dev ffmpeg
!sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev
!sudo apt-get install libxvidcore-dev libx264-dev
!sudo apt-get install libgtk-3-dev
!sudo apt-get install libatlas-base-dev gfortran
%cd /content/drive/MyDrive/treinamento/opencv
!git clone https://github.com/opencv/opencv.git
!git clone https://github.com/opencv/opencv_contrib.git
%cd opencv 
!git checkout 4.6.0 
%cd ..
%cd opencv_contrib 
!git checkout 4.6.0 
%cd /content/drive/MyDrive/treinamento/opencv/opencv
!mkdir build
%cd build
!cmake -D CMAKE_BUILD_TYPE=RELEASE \
 -D CMAKE_INSTALL_PREFIX=/usr/local \
 -D INSTALL_PYTHON_EXAMPLES=OFF \
 -D INSTALL_C_EXAMPLES=ON \
 -D OPENCV_ENABLE_NONFREE=ON \
 -D OPENCV_EXTRA_MODULES_PATH=/content/drive/MyDrive/treinamento/opencv/opencv_contrib/modules \
 -D BUILD_EXAMPLES=ON ..
!make -j4
!sudo make install

Minha unica mudança aqui foi o local de instalação.

Compilando a darknet

Bom... Agora que temos o opencv instalado, vamos instalar a darknet, para isso vamos usar o fork que fiz para não precisar configurar o cmake.

!export OpenCV_DIR="/content/drive/MyDrive/treinamento/opencv/build/"
%cd /content/drive/MyDrive/treinamento
!git clone https://github.com/eduumach/darknet
%cd /content/drive/MyDrive/treinamento/darknet
!make

Para ajudar vou disponibilizar o colab aqui, ai eh só dar um (Ctrl+F9). Link colab

Usando a darknet...

Bom... Agora temos tudo configurado e compilado eh só usar nosso modelo. Para isso vamos precisa somende de rodar esses comandos:

Vamos primeiro montar nosso drive no projeto colab.

from google.colab import drive
drive.mount('/content/drive')

Com esse comando vamos exportar a compilação do opencv:

!export OpenCV_DIR="/content/drive/MyDrive/treinamento/opencv/build/"

Agora com esse vamo colocar a darknet para ser usado em qualquer lugar:

%cd /content/drive/MyDrive/treinamento/darknet
cp darknet /usr/local/bin 

Pronto, tudo configurado agora eh so subir suas anotações para seu drive, dar um %cd para a pasta onde esta, aqui eu reservei dentro de /content/drive/MyDrive/treinamento/treinamentos.

Com isso é só rodar o !darknet <o comando que voce quer usar>.

Vou deixar aqui a documentação da darknet.

Documentação Darknet

Finalizando

Bom... Foi isso espero que gostem e seja util pra voce. Caso queira me conhecer vou deixar minhas redes sociais.

Linkedin

Github