【DenseNet模型】

1 DenseNet结构

在这里插入图片描述
参考链接:https://arxiv.org/pdf/1608.06993.pdf
DenseNet通过密集连接,可以缓解梯度消失问题,加强特征传播,鼓励特征复用,极大减少了参数量。

2 DenseNet特征图保持一致方法

DenseNet的密集连接方式需要特征图大小保持一致
在这里插入图片描述
图2 一个有三个Dense Block的DenseNet网络。两个相邻块之间的层称为过渡层,通过卷积和池化来改变特征图的大小。

3 模型预览方法

densenet121 = torchvision.models.densenet121(pretrained=True)
print(densenet121)

imgs_batch = torch.randn((8, 3, 224, 244))
from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter('my_log/densenet121')
writer.add_graph(densenet121, imgs_batch)
# 在promote中输入tensorboard --logdir path --host=127.0.0.1 ,path为绝对地址,不需要双引号,按照提示打开tensorboard

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值