pip - 如何在Keras中更改预训练模型的默认下载目录？

Question

根据documentation的说法，所有Keras文件的默认下载目录是$HOME/.keras。我正在使用虚拟环境，并且想将预先训练的模型的默认下载目录更改为其他目录。也许这与virtualenv的关系比与Keras的关系更多？

Answer 1

根据文件

Signature: ResNet50(include_top=True, weights='imagenet', input_tensor=None, input_shape=None, pooling=None, classes=1000)

没有参数可以指定将预训练的模型权重下载到何处。

(1)您可以做的是使用 mv( https://www.macworld.com/article/2080814/master-the-command-line-copying-and-moving-files.html)将文件从终端下载后移至所需的位置。

更新:我去检查Keras的github仓库( https://github.com/keras-team/keras/blob/master/keras/applications/resnet50.py)，找到了权重的链接。对于resnet:

WEIGHTS_PATH = 'https://github.com/fchollet/deep-learning-models/releases/download/v0.2/resnet50_weights_tf_dim_ordering_tf_kernels.h5'
WEIGHTS_PATH_NO_TOP = 'https://github.com/fchollet/deep-learning-models/releases/download/v0.2/resnet50_weights_tf_dim_ordering_tf_kernels_notop.h5'

您可以使用任何方法(即 urllib)将这些权重直接下载到文件系统中。