PyRetri: A PyTorch-based Library for Unsupervised Image Retrieval by Deep Convolutional Neural Networks

原文地址:Aggregating Deep Convolutional Features for Image Retrieval

官方实现:PyRetri/PyRetri

摘要

Despite significant progress of applying deep learning methods to the field of content-based image retrieval, there has not been a software library that covers these methods in a unified manner. In order to fill this gap, we introduce PyRetri, an open source library for deep learning based unsupervised image retrieval. The library encapsulates the retrieval process in several stages and provides functionality that covers various prominent methods for each stage. The idea underlying its design is to provide a unified platform for deep learning based image retrieval research, with high usability and extensibility. To the best of our knowledge, this is the first open-source library for unsupervised image retrieval by deep learning.

尽管将深度学习方法应用于基于内容的图像检索领域已经取得了重大进展,但目前还没有一个软件库以统一的方式涵盖这些方法。为了填补这一空白,我们引入了PyRetri,这是一个基于深度学习的无监督图像检索开源库。该库将检索过程封装在几个阶段中,并为每个阶段提供了涵盖各种主要方法的功能。其设计思想是为基于深度学习的图像检索研究提供一个统一的平台,具有高度的可用性和可扩展性。据我们所知,这是第一个通过深度学习进行无监督图像检索的开源库。