前言

今天在Windows系统使用Tensorflow Keras加载inception_resnet_v2卷积神经网络模型。

 feature_extractor = keras.applications.inception_resnet_v2.InceptionResNetV2(
        weights="imagenet",
        include_top=False,
        pooling="avg",
        input_shape=(IMG_SIZE, IMG_SIZE, 3),
    )

运行后报错None – [WinError 10054] 远程主机强迫关闭了一个现有的连接。

具体报错如下：Exception: URL fetch failure on https://storage.googleapis.com/tensorflow/keras-applications/inception_resnet_v2/inception_resnet_v2_weights_tf_dim_ordering_tf_kernels_notop.h5: None – [WinError 10054] 远程主机强迫关闭了一个现有的连接。
科学上网也无法解决这个问题。

解决方法：下载文件至Keras缓存文件夹（C:/Users/用户名/.Keras/models）

程序的执行过程是先检查keras的缓存文件夹里有没有对应文件，如果没有才会去下载，所以可以预先下载预训练模型并放入缓存文件夹，跳过下载的过程。
点击报错的链接，下载在本地之后拷贝到C:/Users/用户名/.Keras/models文件夹（Windows用户！)，问题解决。