在使用Keras的预训练权重做微调、预测时,加载模型的过程中会出现网络中断,博主尝试多次仍没有成功,遂找到下载的地址用浏览器下载,下载地址为:
https://github.com/fchollet/deep-learning-models/releases/
若是下载不成功,多试几次就好,和网速有关。
此外,还有好心人放出基于TensorFlow为backend的模型权重的云盘的链接:
http://pan.baidu.com/s/1dE9giOD
我使用的是Windows下的Anaconda,亲测存放目录为:
C:\Users\电脑用户名\.keras\models
网上还有很多人说以下这个位置是不对的,
D:\Anaconda3\Lib\site-packages\tensorflow\contrib\keras\api\keras\applications\vgg16
Linux下权重文件存放地址为:
~/.keras/models/
这个未曾测试,仅当参考。
如果嫌麻烦,或者不想把权重文件放在C盘,还可以自定义权重文件存放路径,在加载权重的时候用上权重路径即可,举例如下:
model = VGG19(weights='权重文件路径\vgg19_weights_tf_dim_ordering_tf_kernels_notop.h5',include_top=False)
参考:
https://www.cnblogs.com/eniac1946/p/7808439.html