美文网首页
用Netron查看被截取的分类网络层

用Netron查看被截取的分类网络层

作者: LabVIEW_Python | 来源:发表于2021-09-26 11:23 被阅读0次

    在迁移学习中,一般只加载网络的特征提取层,通过Netron可以看到基础网络哪些部分被截取掉了
    范例代码:

    from tensorflow import keras 
    
    # 实例化一个具有预训练权重的基础模型
    base_model = keras.applications.Xception(include_top=True)
    base_model.save("base_model_with_top.h5")
    base_model_without_top = keras.applications.Xception(include_top=False)
    base_model_without_top.save("base_model_without_top.h5")
    
    include_top=True vs include_top=False

    相关文章

      网友评论

          本文标题:用Netron查看被截取的分类网络层

          本文链接:https://www.haomeiwen.com/subject/dnrqnltx.html