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.
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
Abra o site do colab e aperte em "novo notebook". Com isso vai aparecer uma tela parecida com essa:
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.
Finalizando
Bom... Foi isso espero que gostem e seja util pra voce. Caso queira me conhecer vou deixar minhas redes sociais.