reid_demo Namespace Reference

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

Variable Documentation

cfg_path

str reid_demo.cfg_path = '/workspace/configs/reid/mgn/config.yaml'

Definition at line 25 of file reid_demo.py.

checkpoint_path

str reid_demo.checkpoint_path = '/workspace/checkpoints/reid/mgn/market_mgn_R50-ibn.pth'

Definition at line 26 of file reid_demo.py.

crop_image_folder

str reid_demo.crop_image_folder = '/workspace/a_track_result_crop'

Definition at line 27 of file reid_demo.py.

device

str reid_demo.device = 'cuda:1'

Definition at line 24 of file reid_demo.py.

distmat

reid_demo.distmat = get_reid_matrix(feats)

Definition at line 39 of file reid_demo.py.

feats

reid_demo.feats = get_reid_feats(model, file_path)

Definition at line 38 of file reid_demo.py.

file_list

reid_demo.file_list

Definition at line 33 of file reid_demo.py.

file_path

reid_demo.file_path = file_path[0:min(len(file_path), 1000)]

Definition at line 33 of file reid_demo.py.

keep_index

reid_demo.keep_index

Definition at line 42 of file reid_demo.py.

model

reid_demo.model = build_reid_model(cfg_path, checkpoint_path, device)

Definition at line 30 of file reid_demo.py.

s_sim

reid_demo.s_sim

Definition at line 42 of file reid_demo.py.

select_id

reid_demo.select_id

Definition at line 42 of file reid_demo.py.

sorted_keep_index

reid_demo.sorted_keep_index = index_sort(keep_index, s_sim)

Definition at line 43 of file reid_demo.py.

sys_path

list reid_demo.sys_path = ['/workspace/sm_action', '/workspace', '/workspace/smreid']

Definition at line 3 of file reid_demo.py.

threshold

reid_demo.threshold

Definition at line 42 of file reid_demo.py.