big_transfer
big_transfer copied to clipboard
demo: BiT in (poké-)production for neural image search
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.
