reid_demo.py File Reference

Go to the source code of this file.

Namespaces
namespace	reid_demo

Variables
list	reid_demo.sys_path = ['/workspace/sm_action', '/workspace', '/workspace/smreid']
str	reid_demo.device = 'cuda:1'
str	reid_demo.cfg_path = '/workspace/configs/reid/mgn/config.yaml'
str	reid_demo.checkpoint_path = '/workspace/checkpoints/reid/mgn/market_mgn_R50-ibn.pth'
str	reid_demo.crop_image_folder = '/workspace/a_track_result_crop'
	reid_demo.model = build_reid_model(cfg_path, checkpoint_path, device)
	reid_demo.file_list
	reid_demo.file_path = file_path[0:min(len(file_path), 1000)]
	reid_demo.feats = get_reid_feats(model, file_path)
	reid_demo.distmat = get_reid_matrix(feats)
	reid_demo.s_sim
	reid_demo.keep_index
	reid_demo.select_id
	reid_demo.threshold
	reid_demo.sorted_keep_index = index_sort(keep_index, s_sim)