(已解决)keras下载预训练模型中断怎么办

当使用keras下载预训练模型时,经常会中断。比如如下代码:

convBase = keras.applications.MobileNet(weights='imagenet',include_top='False')

此时可以通过下载到本地,再加载的方式,具体步骤:

第一步:修改代码,将上述weights='imagenet'改为weights=None

第二步:下载权重.h5文件,到任意位置

convBase.load_weights('目录/inception_v3_weights_tf_dim_ordering_tf_kernels.h5')

你可能感兴趣的:(keras,python,人工智能)