Resnet模型参数

Question

下面是我为Resnet编写的python代码片段：

base_model = ResNet50(include_top=False, weights='imagenet')

include_top=False, weights='imagenet'的真正含义和用途是什么？

Answer 1

您可以在docs中找到答案

include_top：是否在网络顶部包含全连接层。

权重：None (随机初始化)、'imagenet‘(ImageNet上的预训练)或要加载的权重文件的路径之一。

Answer 2

根据文档：

• include_top：是否在网络顶部包含完全连接层。

• weights：None (随机初始化)、'imagenet‘(ImageNet上的预训练)或要加载的权重文件的路径之一。

在this post中很好地解释了关键字include_top。当涉及到weights时，将其设置为imagenet意味着使用带有ImageNet训练权重的ResNet50架构。您可以使用任何其他数据集进行训练，并使用此参数加载结果权重，或者只使用随机数据集，尽管从预先训练的ImageNet开始可能是一个好主意。由于它们的可移植性，在ImageNet上预先训练的网络通常是一个很好的选择。