caffe2官网教程学习（二）—— 加载预训练模型

来源地址：https://caffe2.ai/docs/tutorial-loading-pre-trained-models.html

1.下载模型

模型库：https://github.com/caffe2/models

#squeezenet为模型名称

python -m caffe2.python.models.download -i squeezenet

或者用Git克隆

git clone https://github.com/caffe2/models

2.导入模块

# where you installed caffe2. Probably '~/pytorch' or '~/src/pytorch'.
CAFFE2_ROOT = "~/pytorch"
# assumes being a subdirectory of caffe2
CAFFE_MODELS = "~/pytorch/caffe2/python/models"
# if you have a mean file, place it in the same dir as the model

%matplotlib inline
from caffe2.proto import caffe2_pb2
import numpy as np
import skimage.io
import skimage.transform
from matplotlib import pyplot
import os
from caffe2.python import core, workspace
import urllib2
print("Required modules imported.")

3.图像传输

其中code是AlexNet object codes

IMAGE_LOCATION =  "https://cdn.pixabay.com/photo/2015/02/10/21/28/flower-631765_1280.jpg"

# What model are we using? You should have already converted or downloaded one.
# format below i