Classificação de raças de gatos com deep learning e TensorFlow

Baixe e Instale dependencias

# Atualiza o comando pip
pip install --upgrade pip

# instala a partir do comando pip o tensorflow
pip install tensorflow

#Instale o pacote de modelos para extrair ao modelo pré treinado mobilenet_v2
pip install tensorflow-hub

#Intale matplotlib
sudo apt-get install python3-matplotlib

ou

pip3 install -U matplotlib

Iniciar o re-treinamento do modelo

python3 retrain.py

Após a execução, o código irá salvar o modelo gerado na pasta retrained/saved_models/cat_breads, que será criada.

Consultar o modelo gerado

python3 classify_cat.py <imagem de entrada>

Próximos passos

Podemos pegar o modelo que treinamos e criar uma aplicação para disponibilizar uma API que recebe a imagem de um gato e retorna sua raça fazendo consulta a este modelo. Ou podemos também criar uma aplicativo para celular, onde tiramos a foto de um gato e consultamos o modelo para saber a sua raça.

Estes são alguns exemplos de como utilizar isso no mundo real, e lembre-se de que podemos utilizar este modelo também para outros tipos de imagens e classificações.

Fonte