Skip to content

ricgoe/YAIR

Repository files navigation

Note

Installation mit conda wird empfohlen. conda env create -f environment.yml Danach git submodule update --init --recursive um submodule threadables zu laden

panda panda-x-tiger

YAIR (pronounced yeah) - Yet Another Image Recommender

Überblick

Im Rahmen des Moduls Big Data Engineering haben wir ein Content-Based Image Recommender System entwickelt. Ziel ist es, für ein Eingabebild die ähnlichsten Bilder aus einem Datensatz effizient zu finden.

Installation/Usage

Download des Index, dem KMeans-Modell sowie der Datenbank
Anpassung der Pfade in GUI.py
Drag'n'Drop Image onto GUI and enjoy! 😜

Features

  • Kombination aus:
    • Farbhistogrammen
    • DINO Embeddings
    • BOVW mit SIFT/ORB-Descriptors
  • Zusammenführung zu einem gewichteten Feature-Vektor
  • FAISS für schnelle Ähnlichkeitssuche mit:
    • Cosine Similarity (durch L2-Normalisierung + Inner Product Index)
  • SQLite zur Verwaltung der Pfade & Metadaten

Pipeline

  1. Feature-Extraktion & Speicherung in SQLite
  2. Aufbau eines FAISS-Index (normalisierte Vektoren)
  3. Query: Extraktion → Normalisierung → Suche im Index
  4. Rückgabe der Top-k ähnlichen Bilder
  5. Anzeige in GUI

code-diagramm

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages