feature.py 文件源码

def multi_im_run(self, image_name):
        """detection and extraction with many boxes"""
        #caffe.set_mode_gpu()
        multi_im = self.detect(image_name, multi_box=True)
        features = []
        for im in multi_im: 
            image = pad(im,size=224)
            feature = extraction.forward(self.net_e, image, self.transformer)
            r = np.squeeze(feature['pool5/7x7_s1'].data[0])
            #feature2 = extraction.forward(self.net_e2, image, self.transformer2)
            #r2 = np.squeeze(feature2['pool5/7x7_s1'].data[0])
            #r = np.hstack((r, r2)).copy()
            #r = r2
            if self.pca is not None:
                r = self.pca.transform(r)[0,:]
            r = r/norm(r)
            #print r.shape
            features.append(r)
        return features