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

Importe as dependências

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.

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.

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

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".

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

__

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).