对于我正在复兴的某个较旧的机器学习项目(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上?
答案 0 :(得分:1)
当然,您需要将文件放入~/.keras/models
中,Keras会自动将其拾取。