1.个人平台说明
ubuntu18.04,RXT3060,pycharm
2.环境配置记录
按照README.md文件依次进行配置(包括激活命名为transt的conda环境),配置完成之后,在工程主目录TransT下面创建一个networks文件夹,然后将下载的预训练模型transt.pth移动到里面。然后修改run_video.py中的参数,然后运行run_video.py。出现以下错误,按顺序进行记录。
2.1.报错“RuntimeError:CUDA error:no kernel image is available for execution on the device
解决方案:
更新pytorch、torchvision的版本。我个人从代码作者推荐的pytorch=1.5 torchvision=0.6.1 cudatoolkit=10.2更新到了torch1.9.1+cu111 torchvision0.10.1+cu111 torchaudio==0.9.1。具体命令如下:
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
想下载其它版本请参考网址:https://pytorch.org/get-started/previous-versions/
2.2.报错“ImportError: cannot import name ‘int_classes’ from ‘torch._six’”
解决方案:
将from torch._six import string_classes, int_classes注释掉,然后添加int_classes = int、string_classes = str,查阅得知原因可能是torch版本和torchvision版本不对应,如果是版本不对应就要换版本了。如果不是因为版本原因报错的,这样修改有效,具体如下:
# from torch._six import string_classes, int_classes
int_classes = int
string_classes = str
2.3.报错“cannot import name ‘_new_empty_tensor’ from ‘torchvision.ops’”
解决方案:
将下面三行代码直接注释掉。可能的原因请参照:http://www.taodudu.cc/news/show-5758046.html?action=onClick
# if float(torchvision.__version__[:3]) < 0.7:
# from torchvision.ops import _new_empty_tensor
# from torchvision.ops.misc import _output_size
3.参考
- MixFormer目标跟踪CVPR2022(代码调试):http://www.taodudu.cc/news/show-5758046.html?action=onClick