Skip to content

Generating embeddings and finding similar images: Run inference on images to get embeddings using EfficientNet and then get K-nearest neighbors to get similar flower images.

Notifications You must be signed in to change notification settings

kumarnikhil936/ImageEmbeddings

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Generating embeddings and finding similar images

  • Download pictures from the image dataset
  • Run inference on them to get embeddings using EfficientNet
  • Show random image, and get K-nearest neighbors to get similar flower images

EfficientNet:

GitHub Repo

Google AI Blog

It is a novel model scaling method that uses a simple yet highly effective compound coefficient to scale up CNNs in a more structured manner. Unlike conventional approaches that arbitrarily scale network dimensions, such as width, depth and resolution, this method uniformly scales each dimension with a fixed set of scaling coefficients.

EfficientNet

EfficientNet Architecture

Dataset:

TensorFlow Flower Dataset

TF Flowers Dataset

FAISS:

Faiss is a library for efficient similarity search and clustering of dense vectors. It contains algorithms that search in sets of vectors of any size, up to ones that possibly do not fit in RAM. It also contains supporting code for evaluation and parameter tuning. Faiss is written in C++ with complete wrappers for Python/numpy. It is developed primarily at Facebook AI Research.

GitHub Repo

About

Generating embeddings and finding similar images: Run inference on images to get embeddings using EfficientNet and then get K-nearest neighbors to get similar flower images.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published