Caffe2 Model Repository
This is a repository for storing pre-trained Caffe2 models. You can use Caffe2 to help you download or install the models on your machine.
Prerequisites
Install Caffe2 with Python bindings.
Download
To download a model folder locally, run
python -m caffe2.python.models.download squeezenet
which will create a folder squeezenet/
containing both an init_net.pb
and exec_net.pb
.
Install
To install a model, run
python -m caffe2.python.models.download -i squeezenet
which will allow later import
s of the model directly in Python:
from caffe2.python.models import squeezenet
print(squeezenet.exec_net.name)