Tensorflow:避免每次运行都下载Inception V3



对于我正在恢复的一个有点旧的机器学习项目(TensorFlow 1.4(,使用了Inception V3模型(demo.py(:

from tensorflow.python.keras.applications.inception_v3 import InceptionV3
model = InceptionV3()

由于旧版本无法与我在Arch Linux上的Python设置一起正常工作,我决定在Docker容器中运行代码。图像构建如下(Dockerfile(:

FROM python:3.5-stretch
RUN pip install tensorflow==1.4.0
ENV WORKDIR=/usr/src/app
WORKDIR $WORKDIR
COPY demo.py $WORKDIR
CMD ["python", "demo.py"]

我按照如下方式构建图像并运行容器:

$ docker build . -t inception_v3
$ docker run inception_v3

当我运行容器时,几秒钟后会出现一个日志条目:

Downloading data from https://github.com/fchollet/deep-learning-models/releases/download/v0.5/inception_v3_weights_tf_dim_ordering_tf_kernels.h5

大约30秒后,程序继续(带有一条暂时不感兴趣的错误消息…(

我的问题是:是否有可能避免下载,并通过另一种机制提供文件?我可以从GitHub下载一次,但我不知道如何在Python代码中提供该文件。

每次下载文件的时间都很烦人;如果这个文件突然间不再在GitHub上了呢?

当然,您需要将文件放入~/.keras/models中,Keras会自动拾取它。

最新更新