search

rom1504 / image_embeddings

Using efficientnet to provide embeddings for retrieval
https://rom1504.github.io/image_embeddings/
MIT License
151 stars 32 forks source link

try to do inference in ui with tensorflow.js and allow the user to provide their own image #16

Open rom1504 opened 4 years ago

rom1504 commented 4 years ago

https://www.tensorflow.org/js/tutorials/conversion/import_keras

rom1504 commented 4 years ago 
from efficientnet.tfkeras import EfficientNetB0
model = EfficientNetB0(weights="imagenet", include_top=False, pooling="avg")
import tensorflowjs as tfjs
tfjs.converters.save_keras_model(model, "model_tfjs/")
anhnhatuel commented 3 years ago

Sorry but I have a question: How to add a new image without embedding all the image?

rom1504 commented 3 years ago

faiss indices have a add method