![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
mmdetection
文章平均质量分 71
broad-sky
很懒~
展开
-
CUDA cudnn tensorrt链接及Jetson板卡pytorch安装链接
cuda + cudnn + tensorrt 对应好的版本。tensorrt官方链接。tensorrt手册。原创 2023-01-17 16:14:17 · 194 阅读 · 0 评论 -
OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.
其中第一个是torch路径下的,第二个是虚拟环境本身路径下的,转到第二个目录下把它剪切到其他路径下备份就好(最好把路径也备份一下)。采用这样默认安装mmcv-full的方式,如果与你环境里的cuda和torch版本不匹配,就容易出现上面报错。其中,{cu_version}、{torch_version}分别对应cuda和torch的版本号。在pycharm里调试程序时可以直接通过在程序前添加这两个语句解决。注意:mmcv-full我这里推荐用1.2.4版本的。原创 2022-09-13 14:11:08 · 56870 阅读 · 30 评论 -
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 · 3155 阅读 · 20 评论 -
SOLOv2-TensorRT
https://github.com/Broad-sky/SOLOv2-TensorRT 如果觉得有用,欢迎Star!原创 2023-03-08 10:16:16 · 1022 阅读 · 4 评论 -
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 · 632 阅读 · 0 评论 -
mmdetection Registry
mmdetection中 Registry类的使用原创 2021-09-26 15:25:57 · 652 阅读 · 0 评论 -
mmdetection multi_apply
mmdetection multi_apply将输入的函数func作用在Iterable每个参数上。cate_pred, kernel_pred=multi_apply(self.forward_single, new_feats, list(range(len(self.seg_num_grids))), eval=eval,upsampled_size=upsampled_size) 在SOLOv2中,将split_feats()函数返回的特征图,依次使用forward_single()进行计算!原创 2021-09-26 12:03:20 · 4020 阅读 · 0 评论