Image Retrieval Python Packages

pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

kapture

kapture is a file format as well as a set of tools for manipulating datasets, and in particular Visual Localization and Structure from Motion data.

kapture-localization

Provide mapping and localization pipelines based on kapture format

gvdb

High-performance distributed vector database with gRPC API and high-dimensional vector support

hypertag

HyperTag - Intuitive Knowledge Management WebApp & CLI for Humans using Deep Learning & Tags

towhee

Towhee is a framework that is dedicated to making neural data processing pipelines simple and fast.

towhee-models

Towhee is a framework that helps you encode your unstructured data into embeddings.

imgsim

The AugNet Python module contains functions for the fast computation of image similarity.

simpleir

[EXTRACT][RETRIEVAL][METRIC]A Simple Framework for Image Retrieval

torchok

Production-oriented Computer Vision models training pipeline for common tasks: classification, segmentation, detection and representation🥤

doraemon-torch

A powerful baseline for image classification, face recognition and image retrieval with Pytorch

deep-image-matching

Multiview matching with deep-learning and hand-crafted local features for COLMAP and other SfM software. Supports high-resolution formats and images with rotations. Both CLI and GUI are supported.

fastreid

SOTA Re-identification Methods and Toolbox

tf-metric-learning

Minimalistic TensorFlow2+ deep metric/similarity learning library with loss functions, miners, and utils as embedding projector.

yandex-ris

A professional reverse image search and crawling tool that uses Yandex's image search engine to find and download similar images.

imdex

Imdex is a library that allows semantic searches over images sets

dinov2-retrieval

A cli program of image retrieval using dinov2

huixiangdou

HuixiangDou: Overcoming Group Chat Scenarios with LLM-based Technical Assistance

pit30m

Development kit for the Pit30M large scale localization dataset

relsim

🍑 relsim: Relational Visual Similarity | pip install relsim 🌍 (CVPR 2026)

towhee-test

No description available

mytowhee

Towhee is a framework that is dedicated to making neural data processing pipelines simple and fast.