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

问题描述 投票:0回答:1

对于我正在复兴的旧机器学习项目(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上?

python docker tensorflow
1个回答
1
投票

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

© www.soinside.com 2019 - 2024. All rights reserved.