MCG-NJU / MixFormer

[CVPR 2022 Oral & TPAMI 2024] MixFormer: End-to-End Tracking with Iterative Mixed Attention
https://arxiv.org/abs/2203.11082
MIT License
448 stars 75 forks source link

跑got10k验证集时出现AttributeError #77

Closed Coronal-yup closed 1 year ago

Coronal-yup commented 1 year ago

您好,首先非常感谢您的工作!我在跑got10k验证集时出现'MixFormerOnline' object has no attribute 'online_sizes',暂时不知道该如何解决?

songtianhui commented 1 year ago

可以提供一下运行的命令和具体报错message

Coronal-yup commented 1 year ago

运行命令:python tracking/test.py mixformer_vit_online baseline_large --dataset got10k_val --threads 32 --num_gpus 2 --params__model mixformer_vit_large_online.pth.tar 具体报错信息:每个序列都会报错一次'MixFormerOnline' object has no attribute 'online_sizes' 以上是问题的详细描述,十分感谢您!

songtianhui commented 1 year ago

因为你测试的 got10k_val,got10k_val 是用在验证而不是测试中的,如果要测试应该使用 got10k_test。

Coronal-yup commented 1 year ago

好的,问题已解决。请问会发布非online的预训练模型吗?

songtianhui commented 1 year ago

非online的就是online模型去掉score module部分,权重是完全一样的。

Coronal-yup commented 1 year ago

可是我在运行python tracking/test.py mixformer_convmae baseline_large --dataset lasot --threads 32 --num_gpus 4 --params__model mixformer_convmae_large_online.pth.tar --params__search_area_scale 4.5命令时出现[Errno 2] No such file or directory: '/YOUR/PRETRAINED/VIT/DIR/convmae_large.pth'报错,不知道是怎么回事呢?

songtianhui commented 1 year ago

这是要在配置文件中设置backbone的预训练模型路径,需要下载官方的convmae_large.pth。 测试的时候应该不用加载backbone的预训练模型,我忘记修改了,在这一行添加参数 network = build_mixformer_convmae(params.cfg, train=False) https://github.com/MCG-NJU/MixFormer/blob/0253ae14446eceecfe5e17a4777e36ed64cffc9b/lib/test/tracker/mixformer_convmae.py#L15

Coronal-yup commented 1 year ago

好的,我们已经有了idea想要在没有x_ot的版本上进行修改构造一个新模型,非常感谢您的工作,希望有问题能再与您交流。