PyRetri: An Open-Source Deep Learning Based Unsupervised Image Retrieval Library Built on PyTorch

Content-based image retrieval (CBIR) uses a representation of visual content to identify relevant images, and it is one of the fundamental research challenges extensively studied in the multimedia community for decades. There has been the importance of powerful features of convolutional neural networks utilized by CBIRs, but the pipeline of deep learning based unsupervised image retrieval is complicated. Despite all these efforts, there has not been a software library that covers these deep learning methods in a unified manner.

This paper presents ‘PyRetri‘ to fill this gap. ‘PyRetri’ is an open-source deep learning based unsupervised image retrieval toolbox built on PyTorch designed for software engineers and researchers. PyRetri is a flexible deep learning based unsupervised image retrieval toolbox designed with simplicity and flexibility in mind.

Key Features

  • Modular Design: There is a breakdown of the deep learning based unsupervised image retrieval into several stages, and users can easily construct an image retrieval pipeline by selecting and combining different modules.
  • Flexible Loading: The framework is able to adjust to load several types of model parameters.
  • Support of Multiple Methods: The framework directly supports several popular methods designed for deep learning based unsupervised image retrieval, which is also suitable for the re-identification of a person.
  • Configuration Search Tool: The configuration search tool helps users to find the optimal retrieval configuration with various hyper-parameters.

🚀 JOIN the fastest ML Subreddit Community




Image Source:

Asif Razzaq is the CEO of Marktechpost Media Inc.. As a visionary entrepreneur and engineer, Asif is committed to harnessing the potential of Artificial Intelligence for social good. His most recent endeavor is the launch of an Artificial Intelligence Media Platform, Marktechpost, which stands out for its in-depth coverage of machine learning and deep learning news that is both technically sound and easily understandable by a wide audience. The platform boasts of over 2 million monthly views, illustrating its popularity among audiences.

Check out to find 100's of Cool AI Tools