前言
DenseNet发表于CVPR 2017,获得最佳论文
论文:Densely Connected Convolutional Networks
PyTorch实现代码:github链接.
论文解释
不重复造轮子,请看一下我舍友博客解释:
【论文复现】DenseNet(2018)
实验结果
热力图纵轴表示源layer,横轴表示当前layer,上三角的热力图表示当前layer对源layer的重用程度。
花分类数据集的实验
数据集包含在前言的github链接中
数据集:3700张花分类数据集
batch_size:16
epochs:60
采用预训练权重
模型采用DenseNet-BC-121
准确率:
损失:
预测效果: