![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 61
broad-sky
很懒~
展开
-
superglue、superpoint C++ TensorRT
motivation(动机)很早以前自从看了superglue、superpoint算法后,一直想把它们利用TensorRT进行加速实现一遍,由于平常太忙一直没有去做这个事情,不过利用周末的时间,终于把代码堆完了~关于SuperGlue里面的细节,可以参看上一篇文章,其中包括最优传输OPT、注意力机制等。图像配准——SuperGlue - 知乎SuperGlue: Learning Feature Matching with Graph Neural Networks SuperGlue特点一般原创 2022-05-24 09:00:40 · 3124 阅读 · 20 评论 -
torch.flatten
torch.flatten(input,start_dim=0,end_dim=-1)→Tensor input(Tensor) – the input tensor. start_dim(int) – the first dim to flatten end_dim(int) – the last dim to flatten 例如数据维度[2,3,4,4]如果flatten(Input, start_dim= 1),则表示从第一维开始打平所有数据直到最...原创 2021-11-09 17:02:06 · 2268 阅读 · 0 评论 -
torch.optim.lr_scheduler.LambdaLR与OneCycleLR
LambdaLR函数接口:LambdaLR(optimizer, lr_lambda, last_epoch=-1, verbose=False)更新策略:其中 是得到的新的学习率,是初始的学习率, λ是通过参数lr_lambda和epoch得到的。参数:optimizer (Optimizer):要更改学习率的优化器;lr_lambda(function or list):可以为一个lambda函数,也可以传入列表;last_epoch (int):最后一个epo..原创 2021-11-01 15:42:32 · 3549 阅读 · 0 评论 -
torch.nonzero()
一维nonzero()返回的是不为零的值在一维向量中的索引。>>> a = torch.randn(4)>>> atensor([-0.7712, 1.6482, 0.0362, 1.6212])>>> b = (a>0.1)>>> b.nonzero()tensor([[1], [3]])>>> btensor([False, True, False, Tru原创 2021-10-26 09:32:09 · 1407 阅读 · 0 评论 -
mmdetection Runner
Runner作用,通过注册的hook在预定义的位点执行自定义函数实现自定义训练流程。注册钩子类注册训练hooks,其原理是将LrUpdaterHook、OptimizerHook、CheckpointHook、IterTimerHook等钩子类注册到hooks中,实则存储在Runner类的self._hooks列表中。 def register_training_hooks(self, lr_config,原创 2021-09-27 16:52:24 · 631 阅读 · 0 评论 -
mmdetection Registry
mmdetection中 Registry类的使用原创 2021-09-26 15:25:57 · 649 阅读 · 0 评论 -
安装apex时,因安装cryptacular,出现mspdb140.dll,无法继续执行代码
直接使用命令pip install apex安装apex时,需要安装包cryptacular,出现问题,"由于找不到mspdb140.dll,无法继续执行代码"尝试方法一、将由于在“C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin\”中存在mspdb140.dll,因此将“C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\bin\”添加至环境变量中,未解决问题!!原创 2021-08-02 13:45:51 · 2102 阅读 · 0 评论