sis

Einfache Bildsuchmaschine

Demo

Workflow

Nachricht

• [2020.06] Aktualisierte viele Teile des Codes für CVPR 2020 Tutorial

Überblick

• Einfache bildbasierte Bildsuchmaschine mit Keras + Flask. Sie können die Suchmaschine starten, indem Sie zwei Python -Skripte ausführen.
• offline.py : Dieses Skript extrahiert ein tiefes Feature aus jedem Datenbankbild. Jedes Merkmal ist eine 4096D-FC6-Aktivierung aus einem VGG16-Modell mit vorgebildeten Bildnoten.
• server.py : Dieses Skript führt einen Webserver aus. Sie können Ihr Abfragebild über eine Flask-Web-Schnittstelle an den Server senden. Der Server findet ähnliche Bilder wie die Abfrage durch einen einfachen linearen Scan.
• GPUs sind nicht erforderlich.
• Getestet auf Ubuntu 18.04 und WSL2 (Ubuntu 20.04)

Links

• Demo
• Kurs bei CVPR2020 [Slides] [Video]
• Projektseite
• Tutorial und Video von @sdkcarlos

Verwendung

git clone https://github.com/matsui528/sis.git
cd sis
pip install -r requirements.txt

# Put your image files (*.jpg) on static/img

# Then fc6 features are extracted and saved on static/feature
# Note that it takes time for the first time because Keras downloads the VGG weights.
python offline.py

# Now you can do the search via localhost:5000
python server.py

Erweitert: Start auf AWS

• Sie können den Suchmaschinenserver auf AWS EC2 problemlos starten. Bitte öffnen Sie zuerst den Port 5000 und starten Sie eine EC2 -Instanz. Beachten Sie, dass Sie eine Sicherheitsgruppe erstellen müssen, so dass der Port 5000 geöffnet wird.
• Eine CPU-Instanz mit mittlerer Ebene reicht aus, z. B. M5.large.
• Nachdem Sie sich bei der Instanz von SSH angemeldet haben, richten Sie bitte die Python-Umgebung (z. B. von Anaconda) ein.
• Führen Sie offline.py und server.py aus.
• Nachdem Sie python server.py ausgeführt haben, können Sie über etwas wie http://ec2-XX-XX-XXX-XXX.us-west-2.compute.amazonaws.com:5000 auf den Server aus Ihrem Browser zugreifen
• (Erweitert) Wenn Sie das System auf sichere Weise bereitstellen möchten, sollten Sie die Suchmaschine mit dem üblichen Webserver, z. B. UWSGI + Nginx, ausführen.
• (Erweitert) Wenn Sie den System serverlos bereitstellen möchten, ist der AWS -Begleiter der richtige Weg.

Zitat

 @misc{sis,
    author = {Yusuke Matsui},
    title = {Simple Image Search Engine},
    howpublished = {url{https://github.com/matsui528/sis}}
}