![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
PyTorch
文章平均质量分 87
图波列夫
这个作者很懒,什么都没留下…
展开
-
torchvision 中的 deform_conv2d
如 DCNv1 和 DCNv2 论文所述,DeformConv 相比常规卷积的参数量和计算量增加不多,但对网络的提升很大。然而,DeformConv 的计算模式并不利于高效实现,给网络带来的开销比纸面数值大:常规卷积可以采用 Implicit GEMM 的形式,非常高效;DeformConv 需要离散访存和插值,增加了 IO 量和内存占用。在 Torchvision 以及其他框架中,DeformConv2d 采用 Explicit GEMM 的方式实现。具体步骤为:deformable_im原创 2023-07-07 17:53:40 · 2680 阅读 · 2 评论 -
PyTorch Design Philosophy
本文档旨在帮助贡献者和模块维护者理解 PyTorch 演化出的高层设计原则。这些规则并不是硬性规定的,而是用来作为指南,以帮助权衡不同的关注点,解决开发 PyTorch 时可能出现的分歧。有关贡献、模块维护以及如何将分歧升级至核心维护者的更多信息,请参见 [ PyTorch Governance](https://pytorch.org/docs/master/community/governance.html)。翻译 2022-11-23 16:58:02 · 146 阅读 · 0 评论 -
DaSiamRPN测试程序分析
此番开源的 DaSiamRPN 代码不包含文章3.3节中的内容(参见知乎评论),同时公布的 SiamRPNBIG.model 相比论文中的模型通道翻倍,计算量增至原有的4倍。因此,速度亦不可按照原文中考量。该项目的实际内容可以看作是 SiamRPNv2。SiamRPN 和 DaSiamRPN 基于 AlexNet,Fast R-CNN 中 RPN 基于 ZF-Net。ZF-Net 是 AlexN...原创 2018-11-13 18:56:07 · 7264 阅读 · 49 评论 -
open-vot:PyTorch 实现 Siamese-FC
open-vot 是一个 基于 python 的目标跟踪库,其中实现了 KCF、SiamFC、GOTURN 等8种算法。该库采用模块化设计,利于不同算法的比较及扩展。本文主要介绍其中的 SiamFC 实现。运行要求代码运行需安装以下依赖包:conda install matplotlib shapelyconda install -c conda-forge tensorboardx ...原创 2018-12-28 20:19:15 · 4264 阅读 · 3 评论 -
SiamMask 测试程序分析
之前分析了 DaSiamRPN 的测试代码,侧重于执行细节。到了 SiamMask,似乎主题应该有所升华。故事的明线为跟踪器构成,暗线为训练流图。相比于 DaSiamRPN,SiamMask 不仅网络结构是现代化的,系统设计也更具匠心。这便于我们一窥其轮廓。SiamMask/models 文件夹下定义了网络的基本架构。SiamMaskFeaturesRPNMask网络工作流为:z_f\x...原创 2019-04-14 11:42:01 · 5740 阅读 · 35 评论 -
THOP: 统计 PyTorch 模型的 FLOPs 和参数量
THOP 是 PyTorch 非常实用的一个第三方库,可以统计模型的 FLOPs 和参数量。使用方法为:from thop import clever_formatfrom thop import profileclass YourModule(nn.Module): # your definitiondef count_your_model(model, x, y): ......原创 2019-07-30 10:15:36 · 36794 阅读 · 11 评论