【深度学习】【Windows系统基于TensorFlow进行迁移学习】TensorFlow加载VGG16Net模型相关事项

VGG16Net迁移学习

  • 问题产生
  • 解决办法
  • 参考文献

问题产生

在利用下述代码对tensorflow预训练模型进行加载时出现报错,原因是当前网络无法下载预训练模型的资源。

PRETRAINING_VGG16 = tf.keras.applications.VGG16\
(
    include_top=False,# 自主训练Dense层
    weights='imagenet',
    input_tensor=Input(shape=IMAGE_SIZE+(3,)),
    input_shape=(224,224,3),
    pooling=None,
    classes=len(CLASSES),# 102
    classifier_activation='softmax'
)

解决办法

为了解决下载失败的问题,可以手动进行下载。
百度网盘链接

https://pan.baidu.com/s/1UUZ5LeKneF_MXyFVtlDCag

提取码apfg
下载完成后会得到一个.h5文件。对于 W i n d o w s Windows Windows操作系统,随后在下图搜索栏中输入%userprofile%
在这里插入图片描述
点击弹出来的系统文件夹,将该.h5文件放入.keras/models文件夹下。
之后,便可以成功对 V G G 16 VGG16 VGG16模型进行加载。

参考文献

[1]https://blog.csdn.net/qq_35436571/article/details/97760098
[2]https://blog.csdn.net/gdengden/article/details/83998275?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-6.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-6.control

你可能感兴趣的:(深度学习)