Article image
Carlos Lima
Carlos Lima19/04/2024 13:23
Compartilhe

Classificando imagens com redes neurais (fastai): Batman, Joker ou WW?

  • #Machine Learning
  • #Python

Importe as dependências

image

Função que usa o buscador DuckDuckGo para procurar imagens com os temas (Batman, Joker e WW), além disso ela salva em um diretório escolhido por mim e apaga arquivos corrompidos.

image

image

Treinar um Rede Neural com fastai é simples, no site deles eles afirma o seguinte:

  • Create appropriate DataLoaders
  • Create a Learner
  • Call a fit method
  • Make predictions or view results.

E é isso que estamos fazendo, definindo o tipo de arquivo (imagens), dividindo os dados, ajustando o tamanho dos arquivos.

image

Feito isso, selecionamos o modelo para visão computacional e damos 'fit' ou seja, aprendemos com os dados.

image

ps: os resultados não são satisfatórios dado o pouco tempo que dediquei a treinar o modelo, claro, podemos dedicar bem mais tempo.

Vamos analisar algumas imagens que foram usadas para o aprendizado o "batch".

image

Agora vamos testar o nosso modelo, carregando uma imagem para testarmos.

image

__

Um detalhe interessante é que, com essa infraestrutura você pode treinar outras coisas que sentir interesse, mudando dois pontos:

  1. Imagens que quer comparar/classificar
  2. Tempo dedicado a aprimorar o tempo, eu usei 3 você pode mais porém cuidado com o overfitting (aprender bem com o treino e não conseguir prever em dados novos).
Compartilhe
Comentários (2)
Carlos Lima
Carlos Lima - 19/04/2024 13:57

Sim, eu acho um máximo. Vou deixar o código para ti. Tu ajustar e faz seus testes haha!

fastai for computer vision

Renan Oliveira
Renan Oliveira - 19/04/2024 13:46

Muito massa Carlos, vou fazer esse script pra brincar um pouco kk