cnnmodel.py 文件源码

def extractFeatures(self):
        if len(self.image) == 0:
            print 'Warning: No image detected. Features not extracted.'
            return None
        else:
            self.net.blobs['data'].reshape(1, 3, self.crop, self.crop)
            self.net.blobs['data'].data[...] = self.transformer.preprocess('data', self.image)
            self.net.forward()
            features = self.net.blobs[self.layer].data.copy()


            features = np.reshape(features, (features.shape[0], -1))[0]

            if cv2.norm(features, cv2.NORM_L2) > 0:
                features = features / cv2.norm(features, cv2.NORM_L2)
            return features.tolist()