pull/1/head
hubenyi 2020-04-17 21:09:57 +08:00
parent 3d6b5aaed6
commit cdbb164741
1 changed files with 3 additions and 1 deletions

View File

@ -2,12 +2,14 @@
## Introduction
PyRetri is an open source deep learning-based image retrieval toolbox based on PyTorch.
PyRetri is a novel deep learning-based image retrieval toolbox based on PyTorch, which is designed for researchers and engineers.
![image](image/overview.png)
### Major Features
Pyretri is a versatile deep learning-based image retrieval toolbox designed with simplicity, flexibility in mind.
- **Modular Design**: We decompose the deep learning-based image retrieval into several stages and one can easily construct a image retrieval pipeline by combining different modules.
- **Flexible Loading**: The toolbox is able to adapt to load several types of model parameters, including parameters with the same keys and shape, parameters with different keys, and parameters with the same keys but different shapes.
- **Support of Multiple Methods**: The toolbox directly supports several popluar methods designed for deep learning-based image retrieval, which are also suitable for person re-identification.