Classificador de Faixas Etárias

Vicente Moraes
Nov 3 · 2 min read
Aplicativo do modelo

Atividade 2 do #cursofastaicisunb

Para essa semana usamos imagens que obtemos do google para criar um modelo e exportá-lo para uma plataforma online.

Escolhi criar uma rede convolucional que classifica pessoas em relação a sua faixa etária com as classes:

[bebê, criança, adolescente, jovem adulto, adulto ou idoso]

Imagens do dataset após limpeza dos dados

Comecei com imagens a partir de queries simples do google, com isso era comum imagens desconectadas com a keyword requerida, como fotos de crianças ao se pesquisar sobre adultos ou imagens de cartoons e desenhos que divergem muito dos padrões do modelo.

Treinei o modelo por 2 estágios antes de modificar o dataset e a partir dos piores resultados limpei o dataset utilizando o método Image Cleaner mostrado no vídeo da lição 2 do fastai.

Análise dos dados e heatmap

Após o treinamento olhei a matriz de confusão e o heatmap da análise de imagens.

O modelo erra ao reconhecer grupos próximos, como adultos e jovens adultos e crianças com adolescentes, o que são erros comuns que também cometemos.

O heatmap das imagens foca principalmente no rosto das pessoas, que era justamente o comportamento que esperava-se do classificador. Entretanto vemos que ainda há algumas discrepâncias no modelo, principalmente com a classe de idosos que tinha uma desproporcionalidade em ambos qualidade e quantidade de dados.

Teste com foto do ator de Esqueceram de Mim — Macauly Caulkin

Finalmente testei o modelo com algumas fotos fora do dataset, como as do ator “Macauly Caulkin” dos filmes Esqueceram de Mim, onde o modelo identificou corretamente as diferentes etapas da vida do artista.

No geral foram pouco mais de 500 imagens e obtive cerca de 67% de acurácia, em versões futuras do problema um dataset melhor balanceado e com menos imagens discrepantes teria resultados ainda melhores para o problema. Você também pode testar o classificador no domínio: classificador disponível pelo Render.

Você também pode ver o código do classificador no meu repositório do github no link: repositório.

Vicente Moraes

Written by

Curso Fastai CIS UnB

Espaço para interagir, criar e compartilhar sua visão da primeira aula do curso

Welcome to a place where words matter. On Medium, smart voices and original ideas take center stage - with no ads in sight. Watch
Follow all the topics you care about, and we’ll deliver the best stories for you to your homepage and inbox. Explore
Get unlimited access to the best stories on Medium — and support writers while you’re at it. Just $5/month. Upgrade