PyRetri：一个基于PyTorch的深度卷积神经网络无监督图像检索库
信息检索
2020-08-06 v2 计算机视觉与模式识别
机器学习
多媒体
摘要
尽管将深度学习方法应用于基于内容的图像检索领域已取得显著进展，但至今仍缺乏一个以统一方式涵盖这些方法的软件库。为了填补这一空白，我们介绍了PyRetri，一个用于基于深度学习的无监督图像检索的开源库。该库将检索过程封装为几个阶段，并提供了涵盖每个阶段多种主流方法的功能。其设计理念是为基于深度学习的图像检索研究提供一个具有高可用性和可扩展性的统一平台。据我们所知，这是第一个用于基于深度学习的无监督图像检索的开源库。
引用
@article{arxiv.2005.02154,
title = {PyRetri: A PyTorch-based Library for Unsupervised Image Retrieval by Deep Convolutional Neural Networks},
author = {Benyi Hu and Ren-Jie Song and Xiu-Shen Wei and Yazhou Yao and Xian-Sheng Hua and Yuehu Liu},
journal= {arXiv preprint arXiv:2005.02154},
year = {2020}
}
备注
Accepted by ACM Multimedia Conference 2020. PyRetri is open-source and available at https://github.com/PyRetri/PyRetri