Colorir o Paquistão com Deep Learning

Bilal Tahir Blocked Unblock Seguir Seguindo 20 de dezembro de 2018

Deep Learning pode ser divertido. Você dá algumas fotos de gatos e cachorros – e pode descobrir quais fotos são de gatos e quais fotos são de cães. Por outro lado, você pode usar técnicas como perda perceptual ou GANs (Generative Adversarial Networks) para transformar cães em gatos, ou seja, transformar uma imagem em outra.

Existem várias aplicações disso. De incrível AI Art como transferir um estilo de pintura para uma foto: https://towardsdatascience.com/artistic-style-transfer-b7566a216431

Para transformar imagens coloridas em preto e branco. Qual foi o que eu fiz. Mas antes de fazermos isso, vamos entender por que isso funciona.

Realmente – como a aprendizagem profunda funciona em um nível básico é mostrar ao modelo um conjunto de dados e depois o resultado desejado (rótulos). E descobre os padrões que mapeiam os dados para os rótulos por conta própria.

Da mesma forma – neste caso, o que podemos fazer é pegar um monte de fotos coloridas e convertê-las em imagens em preto e branco. Uma tarefa comparativamente mais fácil, pois estamos removendo informações em vez de adicioná-las. Você pode fazer isso facilmente em Python usando uma biblioteca como Pillow.

Então agora as fotos coloridas são nossos dados e as imagens correspondentes em preto e branco são nossos rótulos. Podemos, então, treinar o modelo para entender o mapeamento entre esses dois conjuntos de dados e ele aprenderá como uma foto em preto e branco se parece em cores. Depois de treiná-lo, podemos usá-lo para colorir qualquer foto em preto e branco.

E agora a parte divertida começa quando testamos isso!

Eu queria escolher algumas imagens históricas do Paquistão. Existe um incrível grupo no Facebook chamado Vintage Pakistan que publica imagens históricas e é muito legal ver a história do país através dessas imagens. Eu peguei alguns deles para ver como eles sairiam.

Não é ruim. Podemos ver que há algumas colorações estranhas em algumas áreas e o modelo parece gravitar em direção a uma cor parecida a bege, então há espaço para melhorias.

O tom verde na estrada pode ser porque o modelo está ficando pavimento e grama misturados.

Ainda assim – as imagens não parecem meio ruins 🙂

O famoso poeta Faiz Ahmad Faiz em ação

Ele até faz um trabalho decente para fotos antigas que são de qualidade inferior com um tom amarelado.

E claro esta gema…

A primeira bolsa de valores do Paquistão!

Vou continuar brincando e talvez seguir com mais. Deixe-me saber se você tem sugestões de fotos que gostaria de ver coloridas.

Texto original em inglês.