test on CPU

[tommywu1001]

Thanks to the author for the open source code! how to use the CPU for testing?

[songtianhui]

You can just find the code for setting the model's device

MixFormerV2/lib/test/tracker/mixformer2_vit_online.py

Line 20 in 1041438

self.network = network.cuda()

and change it to cpu. And some codes for data processing

MixFormerV2/lib/test/tracker/tracker_utils.py

Line 27 in 1041438

img_tensor = torch.tensor(img_arr).cuda().float().permute((2,0,1)).unsqueeze(dim=0)

.

[tommywu1001]

thank you for your time and reply！ At 2025-01-02 11:27:51, "songtianhui" ***@***.***> wrote: You can just find the code for setting the model's device https://github.com/MCG-NJU/MixFormerV2/blob/10414383d20de5587b74ffcc07f4b5cc2bec2449/lib/test/tracker/mixformer2_vit_online.py#L20 and change it to cpu. And some codes for data processing https://github.com/MCG-NJU/MixFormerV2/blob/10414383d20de5587b74ffcc07f4b5cc2bec2449/lib/test/tracker/tracker_utils.py#L27. — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>