对于我正在恢复的一个有点旧的机器学习项目(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会自动拾取它。