Is AFLink not possible to apply on MOT test sequences ?

[ssbilakeri]

im getting bellow error when the code is run on test sequences, kindly suggest how to run AFLink and GSI on MOT test sequences.

gts = gts[(gts[:, 6] == 1) * (gts[:, 7] == 1)] # 仅考虑“considered" & "pedestrian"
IndexError: index 7 is out of bounds for axis 1 with size 7

[dyhBUPT]

The code line gts = gts[(gts[:, 6] == 1) * (gts[:, 7] == 1)] is in the class LinkData, which is only used for training:

StrongSORT/AFLink/dataset.py

Line 58 in efc79c2

gts = gts[(gts[:, 6] == 1) * (gts[:, 7] == 1)] # 仅考虑“considered" & "pedestrian"

When you want to inference with AFLink, please refer to the following lines to set root='' of LinkData:

StrongSORT/strong_sort.py

Lines 17 to 20 in efc79c2

	if opt.AFLink:
	model = PostLinker()
	model.load_state_dict(torch.load(opt.path_AFLink))
	dataset = LinkData('', '')