Image to Image Search

O SmartSearch é um mecanismo de pesquisa de imagem reversa que encontra imagens semelhantes gerando legendas e comparando essas legendas. Alimentado pelo Tensorflow e Elasticsearch.

ATUALIZAÇÃO : Espelho deste repositório em Codeberg: Pesquisa de imagem a imagem

ATUALIZAÇÃO : APIs REST para gerador de legenda de imagem e pesquisa de imagem são adicionadas! (Grite para xc0d3rz)

Demonstração

Mecanismo de pesquisa de imagem reversa profunda - demos do youtube a funcionalidade principal.

Galhos

Imagem do Docker

Implementação básica

Pacotes necessários:

• Anaconda
• Keras com back -end do TensorFlow (Python 3.6)
• Pesquisa elástica e Elasticsearch-Py (Pesquisa elástica 6.0)

Para mais, consulte os requisitos.txt

Modelos pré-treinados

• Flickr-8k LSTM PESO (FLICKR8K_CNN_LSTM_V1.P)

Faça o download isto e cole a pasta de modelos dentro.

Saída

Pontas

• Instale o Elasticsearch e sempre verifique se o processo de pesquisa elástico está em execução antes de iniciar o Server.py ou index_database.py.

• Em vez de usar a funcionalidade de upload, cole todas as suas imagens na pasta static/img seguida por python index_database.py para indexar todas essas imagens.

• Se você deseja excluir as imagens indexadas, faça sh delete_index.sh

Obrigado

Agradecemos a materializar o CSS pela estética, bem impressa para os tutoriais do Flask e Andrej Karpathy pela maravilhosa biblioteca de legendas. Grite para XC0D3RZ para a inclusão da API REST!