siamfc++解析
lightning980729
这个作者很懒,什么都没留下…
展开
-
siamfc++加时序进行训练
dataloader 原有是图片对,现在应该取4张图片,一张作init,2张作为时序,另外一张作为searchimg。 首先修改videoanalyst/data/sampler/sampler_impl/track_pair_sampler.py 再修改videoanalyst/data/transformer/transformer_impl/random_crop_transformer.py,同时要改的是videoanalyst/data/utils/crop_track_pair.py 再修改vi原创 2020-09-07 10:10:03 · 624 阅读 · 0 评论 -
siamfc++中的trainer的具体实现
trainer = MODULES[name](optimizer, dataloader, monitors) 在videoanalyst/engine/trainer/trainer_base.py看看参数: default_hyper_params = dict( exp_name="default_training", exp_save="snapshots", max_epoch=20, ) def __init__(se原创 2020-09-05 18:14:41 · 465 阅读 · 0 评论 -
siamfc++中的optimizer具体实现
optim 在main/train.py出现了optim模块的使用 # build optimizer optimizer = optim_builder.build(task, task_cfg.optim, model) 在videoanalyst/optim/builder.py定义optim模块,,主要通过optimizer实现,但又分成了两个部分。 optimizer grad_modifier optimizer = optimizer_builder.build(tas原创 2020-09-05 17:00:00 · 318 阅读 · 0 评论 -
python的继承
继承是一种创建新的类的方式,新创建的叫子类,继承的叫父类、超类、基类。 特点:子类可以使用父类的方法,就是函数 举个简单的例子 class grandFather(): print('我是爷爷') class Parent(grandFather): print('我是父类') class SubClass(Parent): print('我是子类') sub = SubClass() #结果: 我是爷爷 # 我是父类 # 我是子类 #注意:类在定义原创 2020-09-04 11:24:08 · 216 阅读 · 0 评论 -
siamfc++中的datapipeline具体实现
datapipeline首先出现在videoanalyst/data/adaptor_dataset.py,这里定义了dataset的创建,继承于torch.utils.data.dataset self.datapipeline = datapipeline_builder.build(self.task, self.cfg,原创 2020-09-03 21:15:08 · 544 阅读 · 0 评论 -
Pytorch分布式训练
接着上一篇写到加载Dataset,这里引进如何把Dataset分布在多卡进行训练 在多卡情况下分布式训练数据的读取用到了这两个代码 torch.nn.parallel.DistributedDataParallel torch.utils.data.distributed.DistributedSampler dataparallel的做法是直接将batch切分到不同的卡。 sampler确保dataloader只会load到整个数据集的一个特定子集的做法。 DistributedSampler就是为每原创 2020-09-03 19:42:25 · 496 阅读 · 0 评论 -
开启torch新篇章:Pytorch创建Dataset,并加载DataLoader
torch.utils.data.Dataset Dataset类是Pytorch中图像数据集中最为重要的一个类,也是Pytorch中所有数据集加载类中应该继承的父类。其中父类中的两个私有成员函数必须被重载,否则将会触发错误提示: 其中__len__应该返回数据集的大小,而__getitem__应该编写支持数据集索引的函数,例如通过dataset[i]可以得到数据集中的第i+1个数据。 在继承了这个Dataset类之后,我们需要实现的核心功能便是__getitem__()函数,getitem()是Pyth原创 2020-09-03 19:25:59 · 5010 阅读 · 0 评论