随笔分类 - 深度学习框架

DPVO 代码剖析

摘要：来自：https://github.com/princeton-vl/DPVO/blob/c0c5a104c9c58663aa9be62c3f125d5b52874f3e/dpvo/altcorr/correlation.py#L33 class PatchLayer(torch.autograd. 阅读全文

posted @ 2024-09-11 15:14 Zenith_Hugh 阅读(132) 评论(0) 推荐(0) 编辑

Python 中的 super(类名, self).__init__() 的含义

摘要：Question pytorch 中定义的神经网络类的 __init__() 中，经常定义 super(类名, self).__init__()，解释下这句话。比如： class TestNN(nn.Module): # 初始化函数 def __init__(self, parm1, ...): 阅读全文

posted @ 2024-09-10 16:51 Zenith_Hugh 阅读(224) 评论(0) 推荐(0) 编辑

libtorch 版本 Pre-cxx11 ABI 和 cxx11 ABI 区别

摘要：libtorch 的 Pre-CXX11 ABI 和 CXX11 ABI 版本主要区别在于它们使用的 C++ ABI（Application Binary Interface）。ABI 定义了编译后的二进制文件在不同模块之间的接口。C++11 引入了一些新的语言特性，导致 ABI 发生变化，因此有两阅读全文

posted @ 2024-08-30 20:03 Zenith_Hugh 阅读(603) 评论(0) 推荐(0) 编辑

pytorch 的 torchvision.datasets.ImageFolder 来自定义数据集

摘要：import torchvision class ClassificationDataset(torchvision.datasets.ImageFolder): """ YOLOv5 Classification Dataset. Arguments root: Dataset path """ 阅读全文

posted @ 2023-06-05 14:07 Zenith_Hugh 阅读(151) 评论(0) 推荐(0) 编辑

paddlepaddle docker 安装实践与人类属性识别实践

摘要：系统：linux centos docker -v # Docker version 23.0.3, build ... 参考链接：https://www.paddlepaddle.org.cn/documentation/docs/zh/install/docker/linux-docker.ht 阅读全文

posted @ 2023-05-12 11:35 Zenith_Hugh 阅读(334) 评论(0) 推荐(0) 编辑

pytorch 笔记之 model.eval() 和 with torch.no_grad()

摘要：model.eval() 负责改变batchnorm、dropout的工作方式，如在eval()模式下，dropout是不工作的。 torch.no_grad() 负责关掉梯度计算，节省eval的时间。只进行inference时，model.eval() 是必须使用的，否则会影响结果准确性。而阅读全文

posted @ 2022-12-20 00:29 Zenith_Hugh 阅读(818) 评论(0) 推荐(1) 编辑

【Pytorch】调用中间层的结果

摘要：在研究 Retinaface 网络结构时候，有个疑惑，作者怎么把 MobileNetV1 的三个 stage 输出分别接到 FPN 上面的，我注意到下面的代码： import torchvision.models._utils as _utils # 使用 _utils.IntermediateLa 阅读全文

posted @ 2022-09-15 18:54 Zenith_Hugh 阅读(231) 评论(0) 推荐(0) 编辑