How do I change the default download directory for pre-trained model in Keras?

Question

According to documentation, it says that the default download directory for all Keras files is $HOME/.keras. I'm using virtual environment and I want to change the default download directory of pre-trained models to a different directory. Maybe this has more to do with virtualenv than with Keras?

Answer 1

If you are using the master branch of keras, you can set the KERAS_HOME environment variable to set the cache directory. If it is not set, cache directory defaults to $HOME/.keras.

export KERAS_HOME="/path/to/keras/dir"

Add the line to your ".bashrc" to set the variable every time you open up a new terminal.

This has not yet been released, you must use the master branch to use this feature.

Answer 2

According to documentation

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

There's no parameter to specify where to download the pre-trained model weights.

(1) What you can do is to move the file to where you want it to be after the download from your terminal using mv (https://www.macworld.com/article/2080814/master-the-command-line-copying-and-moving-files.html).

---

UPDATE: I went to check the github repo of Keras (https://github.com/keras-team/keras/blob/master/keras/applications/resnet50.py) and found the link to the weights. For 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'

You can download those weights directly to your file system using whatever methods (i.e. urllib).

Answer 3

You can copy the model fiel *.h5 that has been downloaded via other ways into the Keras default models directory ~/keras/models.