Advertisement

论文复现DPatch代码解析(附PyTorch语法与知识点补充)

阅读量:

写在前面

近期需要在DPatch源码的基础上进行调整,将其替换为自定义的网络结构,并生成对应的对抗样本。为了便于后续的修改工作,需要投入一定的时间深入理解DPatch源码的内容。因此,我撰写了这篇博客,用于记录学习源码的过程,以便日后查阅和修改。为提升查阅效率,我将自己对源码的理解直接以注释的形式添加到代码中。由于我的主要目标是将darknet.py文件替换为其他网络架构,并利用train.py文件进行模型训练,因此我重点关注这两个文件,并对其中的输入与输出部分进行了详细分析。

Darknet19的整个的网络结构

在这里插入图片描述

darknet.py文件解析

(1)def _make_layers(in_channels, net_cfg)

复制代码
    #这个文件用来定义网络中的每一层,其中的in_channels是每一层的输入通道数,out_channels是输出通道数
    def _make_layers(in_channels, net_cfg):
    layers = []
    #如果net_cfg这个列表中不止一个

全部评论 (0)

还没有任何评论哟~