upload

README.md

@@ -71,3 +71,5 @@ If you use this toolbox in your research, please cite this project.
 
 ```
 
+## Contacts
+

docs/GETTING_STARTED.md

@@ -160,7 +160,7 @@ We basically categorize retrieval process into 4 components.
 
 Here we show how to add your own model to extract features.
 
-1. Create your model file `retrieval_tool_box/models/backbone/backbone_impl/reid_baseline.py`.
+1. Create your model file `pyretri/models/backbone/backbone_impl/reid_baseline.py`.
 
 ```shell
 import torch.nn as nn
@@ -198,7 +198,7 @@ def ft_net():
     return model
 ```
 
-2. Import the module in `retrieval_tool_box/models/backbone/__init__.py`.
+2. Import the module in `pyretri/models/backbone/__init__.py`.
 
 ```shell
 from .backbone_impl.reid_baseline import ft_net

main/extract_feature.py

@@ -5,10 +5,10 @@ import os
 
 import torch
 
-from retrieval_tool_box.config import get_defaults_cfg, setup_cfg
-from retrieval_tool_box.datasets import build_folder, build_loader
-from retrieval_tool_box.models import build_model
-from retrieval_tool_box.extract import build_extract_helper
+from pyretri.config import get_defaults_cfg, setup_cfg
+from pyretri.datasets import build_folder, build_loader
+from pyretri.models import build_model
+from pyretri.extract import build_extract_helper
 
 from torchvision import models
 

main/index.py

@@ -5,9 +5,9 @@ import os
 import pickle
 
 
-from retrieval_tool_box.config import get_defaults_cfg, setup_cfg
-from retrieval_tool_box.index import build_index_helper, feature_loader
-from retrieval_tool_box.evaluate import build_evaluate_helper
+from pyretri.config import get_defaults_cfg, setup_cfg
+from pyretri.index import build_index_helper, feature_loader
+from pyretri.evaluate import build_evaluate_helper
 
 
 def parse_args():

main/make_data_json.py

@@ -2,7 +2,7 @@
 
 import argparse
 
-from retrieval_tool_box.extract import make_data_json
+from pyretri.extract import make_data_json
 
 
 def parse_args():

main/single_index.py

@@ -5,11 +5,11 @@ import os
 from PIL import Image
 import numpy as np
 
-from retrieval_tool_box.config import get_defaults_cfg, setup_cfg
-from retrieval_tool_box.datasets import build_transformers
-from retrieval_tool_box.models import build_model
-from retrieval_tool_box.extract import build_extract_helper
-from retrieval_tool_box.index import build_index_helper, feature_loader
+from pyretri.config import get_defaults_cfg, setup_cfg
+from pyretri.datasets import build_transformers
+from pyretri.models import build_model
+from pyretri.extract import build_extract_helper
+from pyretri.index import build_index_helper, feature_loader
 
 
 def parse_args():

main/split_dataset.py

@@ -2,7 +2,7 @@
 import argparse
 import os
 
-from retrieval_tool_box.extract.utils import split_dataset
+from pyretri.extract.utils import split_dataset
 
 
 def parse_args():