keras导入VGG16下载太慢解决办法

在这里先下载h5文件(notop版本,也就是不包含全连接层1000类别输出的版本)
https://download.csdn.net/download/dongyuxieshen/9972619
如果要下载全模型,则将下列连接复制到迅雷当中直接下载即可。(一定要用迅雷喔)
https://github.com/fchollet/deep-learning-models/releases/download/v0.1/vgg16_weights_tf_dim_ordering_tf_kernels.h5

然后将下载下来的.h5文件其复制到
C:\Users\MJY.keras\models
然后在pycharm里面试验成功:

from keras.applications import VGG16

# weights 指定模型初始化的权重检查点
# include_top 指定模型最后是否包含全连接层,因为这个默认有1000个类别的分类,这里我们只需要识别猫狗,因此不需要包含他
# 如果不填入input_shape也是可以的,网络能够处理任意形状的输入
conv_base = VGG16(weights='imagenet', include_top=False, input_shape=(150, 150, 3))




if __name__ == "__main__":
    print(conv_base.summary())

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