This repo is forked from https://github.com/JDAI-CV/fast-reid and provided some demo-codes based on NNI(https://github.com/microsoft/nni/)
- NNI-HPO , Hyperparameter Optimization
- NNI-NAS, Neural Architecture Search
- NNI-MC, Model Compression, Prune
Firstly, you should install the nni package (visit https://blog.csdn.net/wb3533366/article/details/134978044 for more details )
pip install nni
then:
python nni_tune.py
Multi-GPU is not supported now
python nni_nas.py --config-file ./configs/Market1501/sbs_elan_nas.yml --num-gpus 1
Fastreid is released under the Apache 2.0 license.
If you use FastReID in your research or wish to refer to the baseline results published in the Model Zoo, please use the following BibTeX entry.
@article{he2020fastreid,
title={FastReID: A Pytorch Toolbox for General Instance Re-identification},
author={He, Lingxiao and Liao, Xingyu and Liu, Wu and Liu, Xinchen and Cheng, Peng and Mei, Tao},
journal={arXiv preprint arXiv:2006.02631},
year={2020}
}