demo_track.py 球员id的问题

[Zeyu1226-mt]

您好，我在多个视频上运行demo_track.py时，发现不同视频的球员id是连续的。也就是说，第一个视频最后一个球员id是10的话，下一个视频的第一个球员id是11。请问如何调试代码，使得在跟踪新的视频时，球员id还是从1开始的？期待您的帮助，谢谢您

[ret-1]

Hello @xiyu-ai ,

You could refer to following functions:

MixSort/yolox/evaluators/mot_evaluator.py

Lines 322 to 340 in a078f5b

	for cur_iter, (origin_imgs, imgs, _, info_imgs, ids) in enumerate(
	progress_bar(self.dataloader)
	):
	with torch.no_grad():
	# init tracker
	frame_id = info_imgs[2].item()
	video_id = info_imgs[3].item()
	img_file_name = info_imgs[4]
	video_name = img_file_name[0].split('/')[0]
	if video_name not in video_names:
	video_names[video_id] = video_name
	if frame_id == 1:
	self.args = set_args(self.args, video_name)
	if 'MOT17' in video_name:
	self.args.alpha = 0.8778
	self.args.iou_thresh = 0.2217
	self.args.match_thresh = 0.7986
	tracker.re_init(self.args)

While encountering a new video (L334), tracker.re_init is called to reset the variables.

MixSort/yolox/mixsort_tracker/mixsort_tracker.py

Lines 217 to 221 in a078f5b

	def re_init(self, args, frame_rate=30):
	BaseTrack._count = 0 # set to 0 for new video
	self.tracked_stracks = [] # type: list[STrack]
	self.lost_stracks = [] # type: list[STrack]
	self.removed_stracks = [] # type: list[STrack]

To use our tracker, you need to modify demo_track.py since it remains the same as in original repo.