rom1504/image_embeddings

Provides a complete pipeline for generating and searching image embeddings: downloads datasets from TensorFlow, converts images to TFRecords, runs EfficientNet inference to extract embeddings, and enables approximate nearest-neighbor search via FAISS. The toolkit includes command-line utilities for each step and exports embeddings to numpy/JSON for cross-language compatibility, with an interactive web demo for visualizing semantic image retrieval.

160 stars. No commits in the last 6 months.