Awesome
Ranking Models in Unlabeled New Environments
Prerequisites
This code uses the following libraries
- Python 3.7
- NumPy
- PyTorch 1.7.0 + torchivision 0.8.1
- Sklearn
- Scipy 1.2.1
the environment can be created by using "proxy_set.yml" :
conda env create -f proxy_set.yml
Data Preparation
The folder of each dataset (take Market-1501 as an example) in the data pool should look like this:
Market-1501
├── bounding_box_train/ # the traning set is only necessary for target dataset
│ └── ...
├── bounding_box_test/
│ └── ...
└── query/
└── ...
Run the Code
searching data
python dataset_selection.py --weight 0.6 --result_dir 'sample_data/'
Searched data will be saved in "result_dir". Other parameters, such as the number of clusters, can be set in dataset_selection.py.
Citation
Please cite this paper if it helps your research:
@inproceedings{sun2021,
title={Ranking Models in Unlabeled New Environments},
author={Sun, Xiaoxiao and Hou, Yunzhong and Deng, Weijian and Li, Hongdong and Zheng, Liang},
booktitle={IEEE Conference on International Conference on Computer Vision (ICCV)},
year={2021}
}