Uma ferramenta que utiliza inteligência artificial para criar uma base de links de vídeos. Ele emprega o modelo de rede neural convolucional YOLO para detectar objetos em vídeos do TikTok e registrar sua frequência de ocorrência. Além disso, o projeto também tem como objetivo auxiliar no desenvolvimento de novos modelos de inteligência artificial.
Tela 1920 x 1080 FULL HD: A aplicação requer uma tela com essa resolução para capturar os quadros dos vídeos. O código não é otimizado para resoluções menores.
Placa Nvidia Geforce GTX 1060: Embora seja possível utilizar a CPU, é recomendado utilizar a GPU-CUDA para acelerar a predição dos quadros. É necessário instalar a versão compatível do Torch que corresponda à versão do driver da placa de vídeo. Confira este vídeo para configurar a CUDA na máquina: PyTorch & CUDA Setup - Windows 10. Observações: A versão do CUDA utilizada no projeto é a 12.1. Foi utilizado estes passos para configurar a CUDA na maquina:
- Acesse o site da NVIDIA para instalar o recurso da CUDA 12.1: developer.nvidia;
- Execute o comando "nvidia-smi" no prompt de comando para verificar a versão do driver da placa e a versão do CUDA;
- Acesse o site do PyTorch e instale o recurso da CUDA 12.1;
- Execute o comando "pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121" de acordo com as instruções do site;
Linguagem de programação: O projeto foi desenvolvido em Python 3.9.6. Você pode baixar essa versão aqui.
Banco de dados: Para o gerenciamento do banco de dados, foi utilizado o MySQL Community. Você pode instalar a partir deste link. Na instalação personalizada, selecione apenas o conector do Python.
Configuração das tabelas do banco de dados: Execute o script "instalacao.sql" para criar as tabelas do projeto.
Clonar o projeto: Caso não tenha o Git instalado, você pode baixá-lo aqui. Observações: Para clonar o projeto, siga estes passos:
- Acesse a pasta onde deseja clonar o projeto (por exemplo: "cd C:\tcc");
- Clique com o botão direito do mouse e selecione a opção "Git Bash Here";
- Execute o comando "git clone https://github.com/MCP3H/automacao-tiktok.git" para baixar os arquivos do projeto;
Baixar as dependências do projeto: Utilize o arquivo "requirements.txt" para instalar as bibliotecas necessárias. Se preferir, você pode fazer backup das suas bibliotecas atuais e instalar apenas as do projeto seguindo estes passos:
- Abra o prompt de comando (CMD);
- Navegue até a pasta onde deseja criar o backup das bibliotecas (por exemplo: "cd C:\temp");
- Execute o comando "pip freeze" para mostrar as bibliotecas instaladas atualmente;
- Execute o comando "pip freeze > requirements.txt" para criar um arquivo com as bibliotecas atuais;
- Execute o comando "pip uninstall -r requirements.txt -y" para desinstalar automaticamente as bibliotecas listadas no arquivo gerado;
- Acesse a pasta do projeto clonado (por exemplo: "cd C:\tcc");
- Execute o comando "pip install -r requirements.txt" para instalar as bibliotecas do projeto;
Pré-experimento para definir o critério de aceitação: O arquivo "criterio_aceitacao.py" contém um experimento onde o modelo YOLOv5x avaliou 50 imagens, sendo 40 de cachorros e 10 de espécies parecidas (lobos, coiotes e raposas).
Identificação dos componentes da rede social TikTok: O arquivo "cursor_localizacao.py" foi utilizado para automatizar a aplicação na hora de clicar nas opções na tela, identificando os componentes da rede social.
Análise em tempo real do reconhecimento de objetos: O arquivo "reconhecimento_tempo_real.py" foi criado para verificar em tempo real a qualidade das predições do algoritmo YOLOv5.