Pysot笔记

代码地址:
https://github.com/STVIR/pysot

数据集链接
PySOT中使用的是处理过的图片,经过裁剪后的数据集百度网盘链接:
链接:https://pan.baidu.com/s/1kvEDkNfzgLW7j5PfefC_tA?pwd=wocz
提取码:wocz

训练断点重续
训练过程中不可避免会出现意外,而且训练时间很久,如果重新运行算法,训练会默认重新训练(从epoch1开始),不会断点重续,但其实Pysot代码是有这个功能的,我也是好多次之后才发现。。。
打开tools/train.py文件修改第三百行,指定cfg.TRAIN.RESUME的路径,即可加载之前一半的模型。
在这里插入图片描述

原文链接:https://blog.csdn.net/qq_40199447/article/details/122884138

pretrained_model下载
在这里插入图片描述

从作者提供的谷歌网盘地址下载预训练模型并修改pysot/core/config.py:https://gitcode.net/mirrors/stvir/pysot/-/blob/master/TRAIN.md

训练集数据代表的含义
在这里插入图片描述
路径问题

“/”:代表根目录。

“./”:代表目前所在的目录。

" . ./"代表上一层目录。

参考:http://t.csdn.cn/tvMJ5

项目路径添加到环境变量
在这里插入图片描述
在这里插入图片描述

单机多卡训练
在这里插入图片描述用train.py直接运行时:
单机单卡
在这里插入图片描述
在这里插入图片描述
指定GPU
指定0卡和1卡:os.environ[‘CUDA_VISIBLE_DEVICES’]=“0,1”
但需要将此语句放在import torch之前

os.environ[‘CUDA_VISIBLE_DEVICES’] 是一个环境变量,可以通过设置它来限制程序所能看到的可用 GPU 设备列表,从而确保程序只使用指定的 GPU 设备。设置该环境变量可以使用 os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘0,1’,其中的 ‘0,1’ 表示程序只能看到编号为 0 和 1 的 GPU 设备。

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
批量解压.tar压缩包:for tar in *.tar; do tar xvf $tar; done
在这里插入图片描述

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值