PyRetri下載PyRetri源代碼下載

PyRetri

Python

1.0.0

下載

Pyretri

Benyi Hu，Ren-Jie Song，Xiu-Shen Wei*，Yazhou Yao，Xian-Sheng Hua，Yuehu Liu

通訊作者：xiu-shen wei

介紹

Pyretri（發音為[ˈPerɪˈtriː]）是一種基於統一的深度學習圖像檢索工具箱，該工具箱構建了Pytorch，該工具箱是為研究人員和工程師設計的。

主要功能

Pyretri是一種基於多功能深度學習的無監督圖像檢索工具箱，設計了簡單性和靈活性。

模塊化設計：我們將基於深度學習的無監督圖像檢索分解為多個階段，用戶可以通過選擇和組合不同的模塊來輕鬆構建圖像檢索管道。
靈活加載：工具箱能夠適應加載幾種類型的模型參數，包括具有相同鍵和形狀的參數，帶有不同鍵的參數以及具有相同鍵但形狀不同的參數。
對多種方法的支持：該工具箱直接支持幾種專為基於深度學習的無監督圖像檢索而設計的popluar方法，這也適用於人重新識別。
配置搜索工具：我們提供配置搜索工具，以幫助用戶找到具有各種超參數的最佳檢索配置。

支持的方法

該工具箱支持無監督圖像檢索的Popluar和突出的方法，用戶還可以設計和添加自己的模塊。

預處理
- 直接質量，鋪裝，縮進
- Center Crop，Tencrop
- Twoflip
- Totensor，Tocaffetensor
- 歸一化
功能表示
- 全球平均池（GAP），全球最大池（GMP）
- R-MAC，SPOC，CROW，GEM，SCDA，PWA
- PCB
後處理
- SVD，PCA
- DBA
- QE，K-Reciprocal

更新

2020年5月6日：我們的論文在線，歡迎您在此處下載該論文。

2020年4月29日：我們的Pyretri發射了！

安裝

請參閱install.md以進行安裝和數據集準備。

開始

請參閱get_started.md，以獲取pyretri的基本用法。

模型動物園

結果和模型可在model_zoo.md中找到。

執照

該項目以Apache 2.0許可發布。

引用

如果您在研究中使用此工具箱，請引用本文。

 @inproceedings{Hu2020PyRetri,
  author    = {Benyi Hu and Ren{-}Jie Song and
               Xiu{-}Shen Wei and Yazhou Yao and
               Xian{-}Sheng Hua and Yuehu Liu},
  title     = {PyRetri: {A} PyTorch-based Library for Unsupervised Image Retrieval
               by Deep Convolutional Neural Networks},
  booktitle = {{ACM} International Conference on Multimedia ({ACM MM})},
  pages     = {4461--4464},
  year      = {2020},
}