TransT win10+cuda11.0环境配置

相关链接

代码地址:https://github.com/chenxin-dlut/TransT
陈鑫的bili视频讲解地址:https://www.bilibili.com/video/BV1654y157cS
paper地址:https://arxiv.org/abs/2103.15436

环境配置

git clone git@github.com:chenxin-dlut/TransT.git
  • 创建conda环境
conda create --name transT python=3.7
conda activate transT
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

conda install matplotlib pandas tqdm
pip install opencv-python tb-nightly visdom scikit-image tikzplotlib gdown
conda install cython scipy

pip install pycocotools jpeg4py
pip install wget yacs
pip install shapely==1.6.4.post2

设置环境变量

# Change directory to <PATH_of_TransT>
cd TransT

# Environment settings for pytracking. Saved at pytracking/evaluation/local.py
python -c "from pytracking.evaluation.environment import create_default_local_file; create_default_local_file()"

# Environment settings for ltr. Saved at ltr/admin/local.py
python -c "from ltr.admin.environment import create_default_local_file; create_default_local_file()"

下载模型文件,放在pytracking/networks下即可

运行自己的视频:
脚本:E:\pycharmpro\mysot\TransT\pytracking\run_video.py

parser.add_argument('--tracker_name',default="transt", type=str, help='Name of tracking method.')
parser.add_argument('--tracker_param',default="transt50", type=str, help='Name of parameter file.')
parser.add_argument('--videofile', default=r"your video path",type=str, help='path to a video file.')
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值