Projeto apresentado na matéria de Inteligência Artificial durante minha graduação. Trata-se de um algoritmo de coloração de imagens em escala de cinza. O algoritmo apresentado demonstra uma aplicação de Redes Neurais Convolucionais para a coloração de imagens. Através do aprendizado de máquina, o modelo é capaz de mapear a luminosidade de uma imagem (canal Y) aos seus componentes de cor (canais Cb e Cr), reconstruindo assim a imagem colorida original.
- Instalar os requisitos a partir do arquivo requirements.txt
- realizar a conversão das imagens para o treinamento executando convertion.py
- indicar a imagem que você quer realizar a predição no script main.py
- executar o script main.py
