feats.py 文件源码

def __init__(self, deploy=vgg_deploy, model=vgg_model, mean=vgg_mean, scale_dim=[224, 224], image_dim=[224, 224], isotropic=False):
        caffe.set_mode_gpu()
        caffe.Net.__init__(self, deploy, model, caffe.TEST)

        self.scale_dim = np.array(scale_dim)
        self.image_dim = np.array(image_dim)
        self.isotropic = isotropic

        self.transformer = caffe.io.Transformer({'data':self.blobs['data'].data.shape})
        self.transformer.set_transpose('data', (2,0,1))
        self.transformer.set_mean('data', np.load(mean).mean(1).mean(1))
        self.transformer.set_raw_scale('data', 255)
        self.transformer.set_channel_swap('data', (2,1,0))