google-research / big_transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
https://arxiv.org/abs/1912.11370
Apache License 2.0
1.5k stars 175 forks source link

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

Open hanxiao opened 4 years ago

hanxiao commented 4 years ago

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.