ImageSearcher.py 文件源码

def __init__(self, featurefn):
        self.resultNum = 5
        self.net = caffe.Classifier(model_prototxt, model_trained,
            mean=np.load(mean_path).mean(1).mean(1),
            channel_swap=(2,1,0),
            raw_scale=255,
            image_dims=(256, 256))
        self.parseFeature(featurefn)
        logging.info('Caffe net initialized. Loading cache...')
        #self.buildCache()
        self.loadCache('./featureCache.tsv')
        logging.info('Cache built.')