配置設置:
官方庫鏈接: https://github.com/JDAI-CV/fast-reid#
git clone https://github.com/JDAI-CV/fast-reid.git
安裝依賴:
pip install -r docs/requirements.txt
編譯:切換到fastreid/evaluation/rank_cylib目錄下,執行:
python setup.py install
報錯:
pip install Cython
數據集:
Martket-1501
https://gitcode.com/Universal-Tool/6378f/?utm_source=article_gitcode_universal&index=bottom&type=card&&isLogin=1
上傳至服務器并解壓到datasets文件夾下
將上述三個文件夾解壓到datasets下,
測試環境:
主文件夾下
python tools/train_net.py --config-file ./configs/Market1501/bagtricks_R50.yml MODEL.DEVICE “cuda:0”
可視化結果(demo文件夾):
python demo/visualize_result.py --config-file logs/market1501/bagtricks_R50/config.yaml
–parallel --vis-label --dataset-name Market1501 --output logs/vis_rank_list
–opts MODEL.WEIGHTS logs/market1501/bagtricks_R50/model_final.pth
報錯:
在./fastreid/evaluation/init.py中加一條 from .rank import evaluate_rank
注意:“Market1501”的M必須大寫!
運行完成之后在logs/vis_rank_list下可以看到對應結果