big_transfer icon indicating copy to clipboard operation
big_transfer copied to clipboard

demo: BiT in (poké-)production for neural image search

Open hanxiao opened this issue 5 years ago • 0 comments

Hi there, I made a demo using this BiT model for pokémon search. Result is pretty good, R50x1 gives reasonable performance in terms of efficiency (I set two replicas) and accuracy.

Checkout the code at here https://github.com/jina-ai/examples/tree/master/pokedex-with-bit You can run it directly via Docker, feel free to extend it or try your own photo to see which pokémon are you.

btw, seems that BiT does not work out-of-the-box on tf 2.2.0, I met couple of errors. In the Docker image, I have to set tf version to 2.0.1 to make it work.

hanxiao avatar May 24 '20 00:05 hanxiao