History

liaoxingyu dbf1604231 fix model deploy problems		2021-05-25 15:55:43 +08:00
..
README.md	fix model deploy problems	2021-05-25 15:55:43 +08:00
demo.py	fix model deploy problems	2021-05-25 15:55:43 +08:00
plot_roc_with_pickle.py	onnx/trt support	2020-07-29 17:43:39 +08:00
predictor.py	bugfix for visualize and demo (#468 )	2021-04-21 16:24:34 +08:00
visualize_result.py	bugfix for visualize and demo (#468 )	2021-04-21 16:24:34 +08:00

README.md

FastReID Demo

We provide a command line tool to run a simple demo of builtin models.

You can run this command to get rank visualization results by cosine similarites between different images.

python3 demo/visualize_result.py --config-file logs/dukemtmc/mgn_R50-ibn/config.yaml \
--parallel --vis-label --dataset-name 'DukeMTMC' --output logs/mgn_duke_vis \
--opts MODEL.WEIGHTS logs/dukemtmc/mgn_R50-ibn/model_final.pth

You can also run this command to extract image features.

python3 demo/demo.py --config-file logs/dukemtmc/sbs_R50/config.yaml \
--parallel --input tools/deploy/test_data/*.jpg --output sbs_R50_feat \
--opts MODEL.WEIGHTS logs/dukemtmc/sbs_R50/model_final.pth